minor change

astronomer · Sep 25, 2024 · a8629e6 · a8629e6
1 parent f60c185
commit a8629e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ray_provider/decorators/ray.py b/ray_provider/decorators/ray.py
@@ -31,7 +31,7 @@ class _RayDecoratedOperator(DecoratedOperator, SubmitRayJob):
 
     template_fields: Any = (*SubmitRayJob.template_fields, "op_args", "op_kwargs")
 
-    def __init__(self, config: dict[str, Any] | Callable[[Context], dict[str, Any]], **kwargs: Any) -> None:
+    def __init__(self, config: dict[str, Any] | Callable[..., dict[str, Any]], **kwargs: Any) -> None:
 
         self.config = config
         self.kwargs = kwargs