fix is_prediction function call (#413)

Author: sberan

roboflow/__init__.py

@@ -15,7 +15,7 @@
 from roboflow.models import CLIPModel, GazeModel  # noqa: F401
 from roboflow.util.general import write_line
 
-__version__ = "1.2.8"
+__version__ = "1.2.9"
 
 
 def check_key(api_key, model, notebook, num_retries=0):

roboflow/core/workspace.py

@@ -356,7 +356,6 @@ def _upload_image(imagedesc):
                 sequence_number=imagedesc.get("index"),
                 sequence_size=len(images),
                 num_retry_uploads=num_retries,
-                is_prediction=is_prediction,
             )
 
             return image, upload_time, upload_retry_attempts
@@ -390,6 +389,7 @@ def _save_annotation(image_id, imagedesc):
                 image_id=image_id,
                 job_name=batch_name,
                 num_retry_uploads=num_retries,
+                is_prediction=is_prediction,
             )
 
             return annotation, upload_time

tests/test_project.py

@@ -381,8 +381,8 @@ def test_project_upload_dataset(self):
                 ],
                 "params": {"is_prediction": True},
                 "assertions": {
-                    "upload": {"count": 2, "kwargs": {"is_prediction": True}},
-                    "save_annotation": {"count": 2},
+                    "upload": {"count": 2},
+                    "save_annotation": {"count": 2, "kwargs": {"is_prediction": True}},
                 },
             },
             {
@@ -392,8 +392,8 @@ def test_project_upload_dataset(self):
                 ],
                 "params": {"is_prediction": False},
                 "assertions": {
-                    "upload": {"count": 1, "kwargs": {"is_prediction": False}},
-                    "save_annotation": {"count": 1},
+                    "upload": {"count": 1},
+                    "save_annotation": {"count": 1, "kwargs": {"is_prediction": False}},
                 },
             },
             {
@@ -410,11 +410,18 @@ def test_project_upload_dataset(self):
                     "upload": {
                         "count": 1,
                         "kwargs": {
-                            "is_prediction": True,
                             "batch_name": "prediction-batch",
                             "num_retry_uploads": 2,
                         },
                     },
+                    "save_annotation": {
+                        "count": 1,
+                        "kwargs": {
+                            "is_prediction": True,
+                            "job_name": "prediction-batch",
+                            "num_retry_uploads": 2,
+                        },
+                    },
                 },
             },
         ]