From 6ac006831208af53f72bfb67eec831ed2cbc652d Mon Sep 17 00:00:00 2001
From: Br2850 <bdbristol@gmail.com>
Date: Mon, 18 Nov 2024 01:49:08 -0500
Subject: [PATCH] start enhancements

---
 .../src/components/OperatorExecutionButton/index.tsx       | 3 +++
 fiftyone/operators/types.py                                | 7 +++++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/app/packages/operators/src/components/OperatorExecutionButton/index.tsx b/app/packages/operators/src/components/OperatorExecutionButton/index.tsx
index 53fa289a68..252c536dcc 100644
--- 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
index 89f37fccf5..9baf3f7021 100644
--- 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)