efficient writes and update tests

wayfair-incubator · Sep 5, 2023 · 869e2e7 · 869e2e7
1 parent ff923c3
commit 869e2e7
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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)