update imported tracker name

mikel-brostrom · Oct 1, 2024 · e105138 · e105138
1 parent f112f71
commit e105138
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/performance/test_tracking_p.py b/tests/performance/test_tracking_p.py
@@ -9,10 +9,10 @@
 from boxmot import (
     StrongSort, BotSort, DeepOcSort, OcSort, ByteTrack, ImprAssocTrack, get_tracker_config, create_tracker,
 )
-from tests.test_config import MOTION_ONLY_TRACKING_METHODS, MOTION_N_APPEARANCE_TRACKING_METHODS
+from tests.test_config import MOTION_ONLY_TRACKING_NAMES, MOTION_N_APPEARANCE_TRACKING_NAMES
 
 
-@pytest.mark.parametrize("tracker_type", MOTION_ONLY_TRACKING_METHODS)
+@pytest.mark.parametrize("tracker_type", MOTION_ONLY_TRACKING_NAMES)
 def test_motion_tracker_update_time(tracker_type):
     tracker_conf = get_tracker_config(tracker_type)
     tracker = create_tracker(
@@ -40,7 +40,7 @@ def test_motion_tracker_update_time(tracker_type):
     assert elapsed_time_per_iteration < max_allowed_time, f"Tracking algorithms processing time exceeds the allowed limit:  {elapsed_time_per_iteration} > {max_allowed_time}"
 
 
-@pytest.mark.parametrize("tracker_type", MOTION_N_APPEARANCE_TRACKING_METHODS)
+@pytest.mark.parametrize("tracker_type", MOTION_N_APPEARANCE_TRACKING_NAMES)
 def test_motion_n_appearance_tracker_update_time(tracker_type):
     tracker_conf = get_tracker_config(tracker_type)
     tracker = create_tracker(

diff --git a/tests/test_config.py b/tests/test_config.py
@@ -2,8 +2,8 @@
     StrongSort, BotSort, DeepOcSort, OcSort, ByteTrack, ImprAssocTrack, get_tracker_config, create_tracker,
 )
 
-MOTION_N_APPEARANCE_TRACKING_METHODS = ['botsort', 'deepocsort', 'strongsort', 'imprassoc']
-MOTION_ONLY_TRACKING_METHODS = ['ocsort', 'bytetrack']
+MOTION_N_APPEARANCE_TRACKING_NAMES = ['botsort', 'deepocsort', 'strongsort', 'imprassoc']
+MOTION_ONLY_TRACKING_NAMES = ['ocsort', 'bytetrack']
 
 MOTION_N_APPEARANCE_TRACKING_METHODS=[StrongSort, BotSort, DeepOcSort, ImprAssocTrack]
 MOTION_ONLY_TRACKING_METHODS=[OcSort, ByteTrack]