Added code for a pytest test fix

albumentations-team · Jan 1, 2025 · ef49734 · ef49734
1 parent 4ea7ec2
commit ef49734
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_targets.py b/tests/test_targets.py
@@ -48,6 +48,7 @@ def extract_targets_from_docstring(cls):
 
 DUAL_TARGETS = {
     A.OverlayElements: (Targets.IMAGE, Targets.MASK),
+    A.ConstrainedCoarseDropout : (Targets.IMAGE, Targets.MASK)
 }