Skip to content

Commit

Permalink
add status running to update object
Browse files Browse the repository at this point in the history
Signed-off-by: Yee Hing Tong <wild-endeavor@users.noreply.github.com>
  • Loading branch information
wild-endeavor committed Jan 8, 2025
1 parent c4a53b9 commit 73c0294
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion flytekit/core/worker_queue.py
Original file line number Diff line number Diff line change
Expand Up @@ -202,6 +202,7 @@ def reconcile_one(self, update: Update):
update.status = ItemStatus.RUNNING
else:
if not item.wf_exec.is_done:
update.status = ItemStatus.RUNNING
# Technically a mutating operation, but let's pretend it's not
update.wf_exec = self.remote.sync_execution(item.wf_exec)
if update.wf_exec.closure.phase == WorkflowExecutionPhase.SUCCEEDED:
Expand Down Expand Up @@ -356,7 +357,6 @@ async def add(self, entity: RunnableEntity, input_kwargs: dict[str, typing.Any])
"""
# need to also check to see if the entity has already been registered, and if not, register it.
i = WorkItem(entity=entity, input_kwargs=input_kwargs)
assert i.status == ItemStatus.PENDING

with self.entries_lock:
if entity.name not in self.entries:
Expand Down

0 comments on commit 73c0294

Please sign in to comment.