Fixing tests now that bug is fixed

Author: Diego Ardila

pyproject.toml

@@ -21,7 +21,7 @@ exclude = '''
 
 [tool.poetry]
 name = "scale-nucleus"
-version = "0.1.5"
+version = "0.1.16"
 description = "The official Python client library for Nucleus, the Data Platform for AI"
 license =  "MIT"
 authors = ["Scale AI Nucleus Team <nucleusapi@scaleapi.com>"]
@@ -35,7 +35,7 @@ packages = [{include="nucleus"}]
 python = "^3.6.2"
 grequests = "^0.6.0"
 requests = "^2.25.1"
-tqdm = "^4.60.0"
+tqdm = "^4.41.0"
 dataclasses = { version = "^0.7", python = "^3.6.1, <3.7" }
 
 [tool.poetry.dev-dependencies]
@@ -48,6 +48,11 @@ mypy = "^0.812"
 coverage = "^5.5"
 pre-commit = "^2.12.1"
 
+[tool.pytest.ini_options]
+markers = [
+    "integration: marks tests as slow (deselect with '-m \"not integration\"')",
+]
+
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

tests/test_dataset.py

@@ -305,7 +305,7 @@ def test_annotate_async_with_error(dataset: Dataset):
             "annotation_upload": {
                 "epoch": 1,
                 "total": 2,
-                "errored": 0,  # TODO: fix this once backend fix is in
+                "errored": 1,
                 "ignored": 0,
                 "datasetId": dataset.id,
                 "processed": 1,

tests/test_prediction.py

@@ -283,8 +283,6 @@ def test_mixed_pred_upload_async(model_run: ModelRun):
         asynchronous=True,
     )
     job.sleep_until_complete()
-    print(job.status())
-    print(job.errors())
 
     assert job.status() == {
         "job_id": job.id,
@@ -330,7 +328,7 @@ def test_mixed_pred_upload_async_with_error(model_run: ModelRun):
             "annotation_upload": {
                 "epoch": 1,
                 "total": 2,
-                "errored": 0,
+                "errored": 1,
                 "ignored": 0,
                 "datasetId": model_run._dataset_id,
                 "processed": 1,