allow user to pass callback functions as either operator instance or …

…attribute
GazzolaLab · Nov 20, 2024 · 97a5156 · 97a5156
1 parent 2472b3a
commit 97a5156
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/miv/core/operator/callback.py b/miv/core/operator/callback.py
@@ -96,7 +96,7 @@ def _callback_after_run(self, *args, **kwargs):
             self, "after_run"
         )
         for callback in predefined_callbacks:
-            callback(self, *args, **kwargs)
+            callback(*args, **kwargs)
 
         self._done_flag_after_run = True
 
@@ -122,7 +122,7 @@ def _callback_plot(
 
         plotters = get_methods_from_feature_classes_by_startswith_str(self, "plot_")
         for plotter in plotters:
-            plotter(self, output, inputs, show=show, save_path=save_path)
+            plotter(output, inputs, show=show, save_path=save_path)
         if not show:
             plt.close("all")