start enhancements

voxel51 · Nov 18, 2024 · 6ac0068 · 6ac0068
1 parent 9a13da7
commit 6ac0068
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/app/packages/operators/src/components/OperatorExecutionButton/index.tsx b/app/packages/operators/src/components/OperatorExecutionButton/index.tsx
@@ -21,6 +21,7 @@ export const OperatorExecutionButton = ({
   operatorUri,
   onSuccess,
   onError,
+  onClick,
   executionParams,
   onOptionSelected,
   disabled,
@@ -30,6 +31,7 @@ export const OperatorExecutionButton = ({
   operatorUri: string;
   onSuccess?: ExecutionCallback;
   onError?: ExecutionErrorCallback;
+  onClick?: () => void;
   executionParams?: object;
   onOptionSelected?: (option: OperatorExecutionOption) => void;
   disabled?: boolean;
@@ -38,6 +40,7 @@ export const OperatorExecutionButton = ({
   return (
     <OperatorExecutionTrigger
       operatorUri={operatorUri}
+      onClick={onClick}
       onSuccess={onSuccess}
       onError={onError}
       executionParams={executionParams}

diff --git a/fiftyone/operators/types.py b/fiftyone/operators/types.py
@@ -1400,9 +1400,12 @@ class OperatorExecutionButtonView(Button):
     """
 
     def __init__(self, **kwargs):
-        if "operator" not in kwargs or not isinstance(kwargs["operator"], str):
+        if "operator" not in kwargs or (
+            not isinstance(kwargs["operator"], str)
+            and not callable(kwargs["operator"])
+        ):
             raise ValueError(
-                "The 'operator' parameter of type str is required."
+                "The 'operator' parameter of type str or callable is required."
             )
         super().__init__(**kwargs)