refactor: Changed operator type to BaseOperator in OperatorMethodExec…

…utor
apache · dabla · Sep 29, 2024 · Sep 29, 2024 · Sep 29, 2024 · Sep 30, 2024
commit d013e3d7a2477e5c3b3f32c4e4b623825e2f775c
diff --git a/airflow/models/streamedoperator.py b/airflow/models/streamedoperator.py
@@ -19,7 +19,7 @@
     AirflowException,
     AirflowRescheduleTaskInstanceException,
 )
-from airflow.models import BaseOperator, Operator, TaskInstance
+from airflow.models import BaseOperator, TaskInstance
 from airflow.models.expandinput import (
     ExpandInput,
     DictOfListsExpandInput,
@@ -60,7 +60,7 @@ class OperatorMethodExecutor(LoggingMixin):
     def __init__(
         self,
         semaphore: Semaphore,
-        operator: Operator,
+        operator: BaseOperator,
         context: Context,
         task_instance: TaskInstance,
     ):