changed the check a bit

v7labs · Oct 23, 2023 · 00ceba9 · 00ceba9
1 parent ea1d34e
commit 00ceba9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/darwin/torch/transforms.py b/darwin/torch/transforms.py
@@ -332,7 +332,7 @@ def __call__(self, image, annotation: dict = None) -> tuple:
         transformed_data = self.transform(**albu_data)
         image, transformed_annotation = self._post_process(transformed_data, annotation)
 
-        if list(albu_data.keys()) == ["image"]:
+        if len(annotation) < 1:
             return image
 
         return image, transformed_annotation