diff --git a/dagger/modeler/definition.py b/dagger/modeler/definition.py index 1c0abf3..e2b374b 100644 --- a/dagger/modeler/definition.py +++ b/dagger/modeler/definition.py @@ -545,7 +545,8 @@ async def create_instance( else: if submit_task: template_instance.status = TaskStatus( - code=TaskStatusEnum.SUBMITTED.name, value=TaskStatusEnum.SUBMITTED.value + code=TaskStatusEnum.SUBMITTED.name, + value=TaskStatusEnum.SUBMITTED.value, ) await self.app._store_and_create_task(template_instance) # type: ignore return template_instance diff --git a/dagger/store/stores.py b/dagger/store/stores.py index 7eff0ef..1300cab 100644 --- a/dagger/store/stores.py +++ b/dagger/store/stores.py @@ -168,7 +168,7 @@ async def execute_system_timer_task(self) -> None: # pragma: no cover finished = await task.start(workflow_instance) else: await task.start(workflow_instance) - await self.app._update_instance(task=workflow_instance) # type: ignore + await self.app._update_instance(task=workflow_instance) # type: ignore if finished: await self.remove_trigger(trigger) if not task or task.status.code in TERMINAL_STATUSES: diff --git a/integration_tests/test_app.py b/integration_tests/test_app.py index b516a40..4102984 100644 --- a/integration_tests/test_app.py +++ b/integration_tests/test_app.py @@ -739,7 +739,9 @@ async def simple_data_stream_stop(stream): async for value in stream: instance = await workflow_engine.get_instance(running_task_ids[-1]) - await instance.stop(runtime_parameters=instance.runtime_parameters, workflow_instance=instance) + await instance.stop( + runtime_parameters=instance.runtime_parameters, workflow_instance=instance + ) @workflow_engine.faust_app.agent(simple_topic) @@ -767,7 +769,7 @@ async def simple_data_stream(stream): complete_by_time=120000, repartition=False, seed=rd, - submit_task=True + submit_task=True, ) templates.append(instance) running_task_ids.append(instance.id)