From a8629e6aafdbbc60223ead61af638d7da82656b0 Mon Sep 17 00:00:00 2001 From: Venkat Date: Wed, 25 Sep 2024 13:46:15 -0400 Subject: [PATCH] minor change --- ray_provider/decorators/ray.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ray_provider/decorators/ray.py b/ray_provider/decorators/ray.py index 7acdff5..ae52f57 100644 --- 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