Fix weird typing bug

Author: Ubuntu

nucleus/__init__.py

@@ -792,11 +792,9 @@ def predict(
             for i in range(0, len(other_predictions), batch_size)
         ]
 
-        agg_response = {
-            MODEL_RUN_ID_KEY: model_run_id,
-            PREDICTIONS_PROCESSED_KEY: 0,
-            PREDICTIONS_IGNORED_KEY: 0,
-        }
+        errors = []
+        predictions_processed = 0
+        predictions_ignored = 0
 
         tqdm_batches = self.tqdm_bar(batches)
 
@@ -807,29 +805,27 @@ def predict(
             )
             response = self.make_request(batch_payload, endpoint)
             if STATUS_CODE_KEY in response:
-                agg_response[ERRORS_KEY] = response
+                errors.append(response)
             else:
-                agg_response[PREDICTIONS_PROCESSED_KEY] += response[
-                    PREDICTIONS_PROCESSED_KEY
-                ]
-                agg_response[PREDICTIONS_IGNORED_KEY] += response[
-                    PREDICTIONS_IGNORED_KEY
-                ]
+                predictions_processed += response[PREDICTIONS_PROCESSED_KEY]
+                predictions_ignored += response[PREDICTIONS_IGNORED_KEY]
 
         for s_batch in s_batches:
             payload = construct_segmentation_payload(s_batch, update)
             response = self.make_request(payload, endpoint)
             # pbar.update(1)
             if STATUS_CODE_KEY in response:
-                agg_response[ERRORS_KEY] = response
+                errors.append(response)
             else:
-                agg_response[PREDICTIONS_PROCESSED_KEY] += response[
-                    PREDICTIONS_PROCESSED_KEY
-                ]
-                agg_response[PREDICTIONS_IGNORED_KEY] += response[
-                    PREDICTIONS_IGNORED_KEY
-                ]
+                predictions_processed += response[PREDICTIONS_PROCESSED_KEY]
+                predictions_ignored += response[PREDICTIONS_IGNORED_KEY]
 
+        agg_response = {
+            MODEL_RUN_ID_KEY: model_run_id,
+            PREDICTIONS_PROCESSED_KEY: predictions_ignored,
+            PREDICTIONS_IGNORED_KEY: predictions_processed,
+            ERRORS_KEY: errors,
+        }
         return agg_response
 
     def commit_model_run(

nucleus/model.py

@@ -1,4 +1,4 @@
-from typing import List, Optional, Dict, Type, Union
+from typing import List, Optional, Dict, Union
 from .dataset import Dataset
 from .prediction import (
     BoxPrediction,
@@ -8,14 +8,9 @@
 )
 from .model_run import ModelRun
 from .constants import (
-    ANNOTATIONS_KEY,
-    BOX_TYPE,
-    CUBOID_TYPE,
     NAME_KEY,
-    POLYGON_TYPE,
     REFERENCE_ID_KEY,
     METADATA_KEY,
-    SEGMENTATION_TYPE,
 )