Skip to content

Commit cd14e38

Browse files
committed
Fix ty check for CoiledFunction construction
1 parent dae3351 commit cd14e38

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/pytask_parallel/wrappers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,9 @@ def wrap_task_in_process( # noqa: PLR0913
178178

179179

180180
def rewrap_task_with_coiled_function(task: PTask) -> CoiledFunction:
181-
wrapped = CoiledFunction(wrap_task_in_process, **task.attributes["coiled_kwargs"])
181+
wrapped = cast("Any", CoiledFunction)(
182+
wrap_task_in_process, **task.attributes["coiled_kwargs"]
183+
)
182184
decorated = functools.wraps(wrap_task_in_process)(
183185
cast("Callable[..., Any]", wrapped)
184186
)

0 commit comments

Comments
 (0)