fix(nrtk_transforms): support new return type for perturb()

Kitware · Dec 16, 2024 · a32f4d4 · a32f4d4
1 parent 2eb3d0d
commit a32f4d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nrtk_explorer/library/nrtk_transforms.py b/src/nrtk_explorer/library/nrtk_transforms.py
@@ -192,6 +192,6 @@ def execute(self, input: Image, *input_args: Any) -> Image:
             input_args = ({"img_gsd": 0.15},)
 
         input_array = np.asarray(input)
-        output_array = self._perturber.perturb(input_array, *input_args)
+        output_array, boxes = self._perturber.perturb(input_array, *input_args)
 
         return ImageModule.fromarray(output_array)