increase timeout

SciPhi-AI · Oct 21, 2024 · 0219528 · 0219528
1 parent 75103e0
commit 0219528
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/py/tests/integration/runner_cli.py b/py/tests/integration/runner_cli.py
@@ -253,8 +253,8 @@ def test_kg_create_graph_sample_file_cli():
     print("Calling `poetry run r2r create-graph --run`")
     output = run_command("poetry run r2r create-graph --run")
 
-    # if "queued" in output:
-    time.sleep(60)
+    if "queued" in output:
+        time.sleep(120)
 
     response = requests.get(
         "http://localhost:7272/v2/entities/",

diff --git a/py/tests/integration/runner_sdk.py b/py/tests/integration/runner_sdk.py
@@ -689,7 +689,7 @@ def test_kg_create_graph_sample_file_sdk():
     print(create_graph_result)
 
     if "queued" in create_graph_result["results"]["message"]:
-        time.sleep(60)
+        time.sleep(120)
 
     result = client.get_entities(
         collection_id="122fdf6a-e116-546b-a8f6-e4cb2e2c0a09",