update

astronomer · Sep 19, 2024 · fef3233 · fef3233
1 parent 3767ade
commit fef3233
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ray_provider/decorators/ray.py b/ray_provider/decorators/ray.py
@@ -160,7 +160,7 @@ def task(
         if config is None:
             config = {}
         elif callable(config):
-            config = config()
+            config = config(**kwargs)
         elif not isinstance(config, dict):
             raise TypeError("config must be either a callable, a dictionary, or None")