Skip to content

Commit a24044d

Browse files
committed
Removed probably unneeded configs to Prefect
1 parent d10395b commit a24044d

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/unifair/engine/prefect.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,21 +54,20 @@ def _run_task(self, state: PrefectTask, task: IsTask, call_func: Callable, *args
5454
_prefect_task = state
5555

5656
if task.in_flow_context:
57-
return _prefect_task(*args, return_state=False, **kwargs)
57+
return _prefect_task(*args, **kwargs)
5858
else:
5959
flow_kwargs = dict(name=task.name)
6060

6161
if asyncio.iscoroutinefunction(_prefect_task):
6262

6363
@prefect_flow(**flow_kwargs)
6464
async def task_flow(*inner_args, **inner_kwargs):
65-
return await resolve(
66-
_prefect_task(*inner_args, return_state=False, **inner_kwargs))
65+
return await resolve(_prefect_task(*inner_args, **inner_kwargs))
6766
else:
6867

6968
@prefect_flow(**flow_kwargs)
7069
def task_flow(*inner_args, **inner_kwargs):
71-
return _prefect_task(*inner_args, return_state=False, **inner_kwargs)
70+
return _prefect_task(*inner_args, **inner_kwargs)
7271

7372
return task_flow(*args, **kwargs)
7473

0 commit comments

Comments
 (0)