test upping timeouts

Striveworks · Dec 22, 2023 · dcfc131 · dcfc131
1 parent 02c1d45
commit dcfc131
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/integration_tests/client/datatype/test_data_generation.py b/integration_tests/client/datatype/test_data_generation.py
@@ -101,7 +101,7 @@ def test_generate_prediction_data(client: Client):
             Annotation.type == AnnotationType.BOX,
         ],
     )
-    eval_results = eval_job.wait_for_completion(timeout=3)
+    eval_results = eval_job.wait_for_completion(timeout=30)
     assert eval_results.status == JobStatus.DONE
 
     eval_dict = asdict(eval_results)

diff --git a/integration_tests/conftest.py b/integration_tests/conftest.py
@@ -50,13 +50,13 @@ def db() -> Session:
 
     for model in client.get_models():
         try:
-            client.delete_model(model["name"], timeout=5)
+            client.delete_model(model["name"], timeout=30)
         except exceptions.ModelDoesNotExistError:
             continue
 
     for dataset in client.get_datasets():
         try:
-            client.delete_dataset(dataset["name"], timeout=5)
+            client.delete_dataset(dataset["name"], timeout=30)
         except exceptions.DatasetDoesNotExistError:
             continue