remove Flip augmentation that does not exist in the registry

Author: dtronmans

luxonis_ml/data/augmentations/albumentations_engine.py

@@ -243,13 +243,12 @@ def apply_to_array(
 
     def _check_augmentation_warnings(
         self, config_item: dict[str, Any], available_target_types: set
-    ) -> bool:
+    ) -> None:
         augmentation_name = config_item["name"]
 
         if "keypoints" in available_target_types and augmentation_name in [
             "HorizontalFlip",
             "VerticalFlip",
-            "Flip",
         ]:
             logger.warning(
                 f"Using '{augmentation_name}' with keypoints."
@@ -359,10 +358,11 @@ def __init__(
         available_target_types = set(self.targets.values())
 
         for config_item in config:
+            cfg = AlbumentationConfigItem(**config_item)
+
             self._check_augmentation_warnings(
                 config_item, available_target_types
             )
-            cfg = AlbumentationConfigItem(**config_item)
 
             transform = self.create_transformation(cfg)
 

tests/conftest.py

@@ -125,7 +125,6 @@ def augmentation_config() -> list[Params]:
         {"name": "MixUp", "params": {"p": 1.0}},
         {"name": "Defocus", "params": {"p": 1.0}},
         {"name": "Sharpen", "params": {"p": 1.0}},
-        {"name": "Flip", "params": {"p": 1.0}},
         {"name": "RandomRotate90", "params": {"p": 1.0}},
     ]
 

tests/test_data/test_augmentations/test_special.py

@@ -39,9 +39,6 @@ def test_skip_augmentations():
         {
             "name": "Perspective",
         },
-        {
-            "name": "Flip",
-        },
         {
             "name": "HorizontalFlip",
         },
@@ -87,10 +84,14 @@ def test_skip_augmentations():
     batched_transform_names = [
         t.__class__.__name__ for t in augmentations.batch_transform.transforms
     ]
-
+    print(spatial_transform_names)
     assert spatial_transform_names == [
         "Perspective",
         "Lambda",
+        "HorizontalFlip",
+        "Lambda",
+        "VerticalFlip",
+        "Lambda",
         "Rotate",
         "Lambda",
     ]