fix: a use_task with dependencies=None (user called) can have argument

maartenbreddels · maartenbreddels · commit da2f8cfed15c · 2025-05-16T09:40:36.000+02:00
diff --git a/solara/tasks.py b/solara/tasks.py
@@ -868,46 +868,46 @@ def use_task(
     dependencies: Literal[None] = ...,
     raise_error=...,
     prefer_threaded=...,
-) -> Callable[[Callable[[], R]], "Task[[], R]"]: ...
+) -> Callable[[Callable[P, R]], "Task[P, R]"]: ...
 
 
 @overload
 def use_task(
-    f: Callable[[], R],
+    f: None = None,
     *,
-    dependencies: Literal[None] = ...,
+    dependencies: List = ...,
     raise_error=...,
     prefer_threaded=...,
-) -> "Task[[], R]": ...
+) -> Callable[[Callable[[], R]], "Task[[], R]"]: ...
 
 
 @overload
 def use_task(
-    f: None = None,
+    f: Callable[[], R],
     *,
     dependencies: List = ...,
     raise_error=...,
     prefer_threaded=...,
-) -> Callable[[Callable[[], R]], "Task[[], R]"]: ...
+) -> "Task[[], R]": ...
 
 
 @overload
 def use_task(
-    f: Callable[[], R],
+    f: Callable[P, R],
     *,
-    dependencies: List = ...,
+    dependencies: Literal[None] = ...,
     raise_error=...,
     prefer_threaded=...,
-) -> "Task[[], R]": ...
+) -> "Task[P, R]": ...
 
 
 def use_task(
-    f: Union[None, Callable[[], R]] = None,
+    f: Union[None, Callable[P, R]] = None,
     *,
     dependencies: Union[None, List] = [],
     raise_error=True,
     prefer_threaded=True,
-) -> Union[Callable[[Callable[[], R]], "Task[[], R]"], "Task[[], R]"]:
+) -> Union[Callable[[Callable[P, R]], "Task[P, R]"], "Task[P, R]"]:
     """A hook that runs a function or coroutine function as a task and returns the result.
 
     Allows you to run code in the background, with the UI available to the user. This is useful for long running tasks,
