re-implement timeouts

Striveworks · Dec 22, 2023 · eed6d89 · eed6d89
1 parent 3331904
commit eed6d89
Showing 1 changed file with 2 additions and 2 deletions.
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"])
+            client.delete_model(model["name"], timeout=360)
         except exceptions.ModelDoesNotExistError:
             continue
 
     for dataset in client.get_datasets():
         try:
-            client.delete_dataset(dataset["name"])
+            client.delete_dataset(dataset["name"], timeout=360)
         except exceptions.DatasetDoesNotExistError:
             continue