Fix interop by adding missing arguments

sparkfish · Jul 17, 2024 · ce6ce36 · ce6ce36
1 parent 4132722
commit ce6ce36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/augraphy/utilities/interop.py b/augraphy/utilities/interop.py
@@ -22,7 +22,7 @@ def __init__(self, augmentation, p=1):
         self.augmentation = augmentation
         super().__init__(p=p)
 
-    def __call__(self, image, layer=None, force=False):
+    def __call__(self, image, layer=None, mask=None, keypoints=None, bounding_boxes=None, force=False):
 
         image = image.copy()
         result = self.augmentation(image=image)