add request timeout of 2 minutes

BlaiseMuhirwa · Dec 17, 2023 · 6cbc347 · 6cbc347
1 parent 039c57b
commit 6cbc347
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/.github/workflows/cicd.yaml b/.github/workflows/cicd.yaml
@@ -28,6 +28,10 @@ jobs:
       with:
         version: ${{ env.POETRY_VERSION }}
 
+    - name: Install libomp (MacOS)
+      if: runner.os == 'macOS'
+      run: brew install libomp 
+
     - name: Install dependencies
       run: |
         cd flatnav_python

diff --git a/flatnav_python/unit_tests/test_index.py b/flatnav_python/unit_tests/test_index.py
@@ -131,7 +131,7 @@ def run_test(
     index.add(data=training_set, ef_construction=ef_construction)
     end = time.time()
 
-    print(f"Indexing time = {end - start} seconds")
+    print(f"\nIndexing time = {end - start} seconds")
 
     if use_reordering:
         if not reordering_algorithm:

diff --git a/flatnav_python/unit_tests/test_utils.py b/flatnav_python/unit_tests/test_utils.py
@@ -34,7 +34,7 @@ def get_ann_benchmark_dataset(dataset_name):
     dataset_uri = f"{base_uri}/{dataset_name}.hdf5"
 
     with tempfile.TemporaryDirectory() as tmp:
-        response = requests.get(dataset_uri)
+        response = requests.get(dataset_uri, timeout=120)
         loc = os.path.join(tmp, dataset_name)
 
         with open(loc, "wb") as f: