From ff771ad37177c652defd4e602f1503c104351234 Mon Sep 17 00:00:00 2001 From: HonzaCuhel Date: Thu, 24 Oct 2024 23:07:11 +0200 Subject: [PATCH] Fix: annotator tests --- tests/core_tests/unittests/test_annotators.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/core_tests/unittests/test_annotators.py b/tests/core_tests/unittests/test_annotators.py index 48964fa..7bd024b 100644 --- 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)