Fix incorrect tests

Author: Ubuntu

tests/test_dataset.py

@@ -417,8 +417,8 @@ def test_annotate_async_with_error(dataset: Dataset):
                 "processed": 1,
             },
         },
-        "job_progress": "0.67",
-        "completed_steps": 2,
+        "job_progress": "1.00",
+        "completed_steps": 3,
         "total_steps": 3,
     }
 

tests/test_prediction.py

@@ -348,8 +348,8 @@ def test_mixed_pred_upload_async_with_error(model_run: ModelRun):
                 "processed": 1,
             },
         },
-        "job_progress": "0.67",
-        "completed_steps": 2,
+        "job_progress": "1.00",
+        "completed_steps": 3,
         "total_steps": 3,
     }