Update common.py

Aleph-Alpha · Sep 25, 2024 · 5b70c26 · 5b70c26
1 parent 042ae73
commit 5b70c26
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/common.py b/tests/common.py
@@ -12,7 +12,7 @@ def sync_client() -> Client:
         token=get_env_var("TEST_TOKEN"),
         host=get_env_var("TEST_API_URL"),
         # This will retry after [0.0, 0.25, 0.5, 1.0, 2.0] seconds
-        total_retries=os.environ.get("TEST_API_RETRIES", 5),
+        total_retries=int(os.environ.get("TEST_API_RETRIES", "5")),
     )
 
 
@@ -22,7 +22,7 @@ async def async_client() -> AsyncIterable[AsyncClient]:
         token=get_env_var("TEST_TOKEN"),
         host=get_env_var("TEST_API_URL"),
         # This will retry after [0.0, 0.25, 0.5, 1.0, 2.0] seconds
-        total_retries=os.environ.get("TEST_API_RETRIES", 5),
+        total_retries=int(os.environ.get("TEST_API_RETRIES", "5")),
     ) as client:
         yield client