Fix mypy error

Author: adamtheturtle

src/vws/vws.py

@@ -272,7 +272,7 @@ def _wait_for_target_processed(
         """
         while True:
             report = self.get_target_summary_report(target_id=target_id)
-            if report['status'] != 'processing':
+            if report.status != TargetStatuses.PROCESSING:
                 return
 
             sleep(seconds_between_requests)
@@ -350,10 +350,7 @@ def list_targets(self) -> List[str]:
 
         return list(response.json()['results'])
 
-    def get_target_summary_report(
-        self,
-        target_id: str,
-    ) -> Dict[str, Union[str, int]]:
+    def get_target_summary_report(self, target_id: str) -> TargetSummaryReport:
         """
         Get a summary report for a target.
 

tests/test_vws.py

@@ -300,10 +300,10 @@ def test_wait_for_target_processed(
             application_metadata=None,
         )
         report = vws_client.get_target_summary_report(target_id=target_id)
-        assert report['status'] == 'processing'
+        assert report.status == TargetStatuses.PROCESSING
         vws_client.wait_for_target_processed(target_id=target_id)
         report = vws_client.get_target_summary_report(target_id=target_id)
-        assert report['status'] != 'processing'
+        assert report.status != TargetStatuses.PROCESSING
 
     def test_default_seconds_between_requests(
         self,
@@ -424,7 +424,7 @@ def test_custom_timeout(
             )
 
             report = vws_client.get_target_summary_report(target_id=target_id)
-            assert report['status'] == 'processing'
+            assert report.status == TargetStatuses.PROCESSING
             with pytest.raises(TargetProcessingTimeout):
                 vws_client.wait_for_target_processed(
                     target_id=target_id,
@@ -436,7 +436,7 @@ def test_custom_timeout(
                 timeout_seconds=0.5,
             )
             report = vws_client.get_target_summary_report(target_id=target_id)
-            assert report['status'] != 'processing'
+            assert report.status != TargetStatuses.PROCESSING
 
 
 class TestGetDuplicateTargets: