Aug test fix

mlcommons · Jul 1, 2024 · c312e24 · c312e24
1 parent b423f64
commit c312e24
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/GANDLF/data/augmentation/__init__.py b/GANDLF/data/augmentation/__init__.py
@@ -68,11 +68,12 @@ def get_augmentation_transforms(
 
         if augmentation_type_lower in global_augs_dict:
             current_augmentations.append(
-                global_augs_dict[augmentation_type_lower](**augmentation_params)
+                global_augs_dict[augmentation_type_lower](augmentation_params)
             )
         else:
             warn(
                 f"Augmentation {augmentation_type} not found in the global augmentation dictionary.",
                 UserWarning,
             )
+
     return current_augmentations