Bring back old supervision pin

roboflow · Nov 14, 2024 · 87a69cb · 87a69cb
1 parent d11b5f7
commit 87a69cb
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 12 deletions.
diff --git a/requirements/_requirements.txt b/requirements/_requirements.txt
@@ -11,7 +11,7 @@ prometheus-fastapi-instrumentator~=7.0.0
 redis~=5.0.0
 requests>=2.26.0,<2.32.0  # newer requests breaks docker which would need to be bumped to 7.x.x
 rich~=13.0.0
-supervision~=0.25.0
+supervision>=0.21.0,<=0.22.0
 pybase64~=1.0.0
 scikit-image>=0.19.0,<=0.24.0
 requests-toolbelt~=1.0.0

diff --git a/requirements/requirements.cli.txt b/requirements/requirements.cli.txt
@@ -3,7 +3,7 @@ docker==6.1.3
 typer>=0.9.0,<=0.12.5
 rich~=13.0.0
 PyYAML~=6.0.0
-supervision~=0.25.0
+supervision>=0.21.0,<=0.22.0
 opencv-python>=4.8.1.78,<=4.10.0.84
 tqdm>=4.0.0,<5.0.0
 GPUtil~=1.4.0

diff --git a/requirements/requirements.sdk.http.txt b/requirements/requirements.sdk.http.txt
@@ -2,7 +2,7 @@ requests>=2.26.0,<2.32.0  # newer requests breaks docker which would need to be
 dataclasses-json~=0.6.0
 opencv-python>=4.8.1.78,<=4.10.0.84
 pillow>=9.0.0,<11.0
-supervision~=0.25.0
+supervision>=0.21.0,<=0.22.0
 numpy<=1.26.4
 aiohttp>=3.9.0,<=3.10.11
 backoff~=2.2.0

diff --git a/requirements/requirements.test.unit.txt b/requirements/requirements.test.unit.txt
@@ -10,4 +10,4 @@ pytest-timeout>=2.2.0
 httpx
 uvicorn<=0.22.0
 aioresponses>=0.7.6
-supervision~=0.25.0
+supervision>=0.21.0,<=0.22.0
diff --git a/tests/workflows/integration_tests/execution/test_workflow_with_sahi.py b/tests/workflows/integration_tests/execution/test_workflow_with_sahi.py
@@ -352,7 +352,7 @@ def slicer_callback(image_slice: np.ndarray):
         callback=slicer_callback,
         slice_wh=(640, 640),
         overlap_ratio_wh=(0.2, 0.2),
-        overlap_filter=sv.OverlapFilter.NON_MAX_SUPPRESSION,
+        overlap_filter_strategy=sv.OverlapFilter.NON_MAX_SUPPRESSION,
         iou_threshold=0.3,
     )
 

diff --git a/tests/workflows/integration_tests/execution/test_workflow_with_video_metadata_processing.py b/tests/workflows/integration_tests/execution/test_workflow_with_video_metadata_processing.py
@@ -214,10 +214,7 @@ def test_workflow_with_tracker(
         first_crowd_frame_tracker_ids == second_crowd_frame_tracker_ids
     ), "The same image, expected no tracker IDs change"
     assert first_license_plate_frame_tracker_ids == [
-        1,
-        2,
-        3,
-    ], (
-        "Since `supervision>=0.25.0` tracker IDs are unique for each new tracker instance - and we "
-        "expect new tracker for `metadata_license_plate_image` to be created - hence fresh tracker ids"
-    )
+        15,
+        16,
+        17,
+    ], "External IDs for all trackers are global, hence we offset by numer of all ever generated tracker IDs"