Fix: annotator tests

luxonis · Oct 24, 2024 · ff771ad · ff771ad
1 parent 04e91fd
commit ff771ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/core_tests/unittests/test_annotators.py b/tests/core_tests/unittests/test_annotators.py
@@ -8,8 +8,8 @@
 from PIL import Image
 
 from datadreamer.dataset_annotation.clip_annotator import CLIPAnnotator
-from datadreamer.dataset_annotation.fastsam_annotator import SlimSAMAnnotator
 from datadreamer.dataset_annotation.owlv2_annotator import OWLv2Annotator
+from datadreamer.dataset_annotation.slimsam_annotator import SlimSAMAnnotator
 
 # Get the total disk space in GB
 total_disk_space = psutil.disk_usage("/").total / (1024**3)