Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Mar 11, 2025
1 parent e745bd7 commit f940d33
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 5 deletions.
10 changes: 7 additions & 3 deletions backend/workflow_manager/workflow_v2/execution.py
Original file line number Diff line number Diff line change
Expand Up @@ -290,7 +290,11 @@ def publish_initial_workflow_logs(self, total_files: int) -> None:
)

def publish_final_workflow_logs(
self, total_files: int, successful_files: int, failed_files: int, execution_id: str
self,
total_files: int,
successful_files: int,
failed_files: int,
execution_id: str,
) -> None:
"""Publishes the final logs for the workflow.
Expand All @@ -300,10 +304,10 @@ def publish_final_workflow_logs(
self.publish_average_cost_log(
execution_id=execution_id, total_files=successful_files
)

# To not associate final logs with a file execution
self.file_execution_id = None

self.publish_update_log(LogState.END_WORKFLOW, "1", LogComponent.STATUS_BAR)
self.publish_update_log(
LogState.SUCCESS, "Executed successfully", LogComponent.WORKFLOW
Expand Down
4 changes: 2 additions & 2 deletions backend/workflow_manager/workflow_v2/workflow_helper.py
Original file line number Diff line number Diff line change
Expand Up @@ -213,12 +213,12 @@ def process_input_files(
execution_service.update_execution(ExecutionStatus.COMPLETED)

workflow_execution = execution_service.get_execution_instance()

execution_service.publish_final_workflow_logs(
total_files=total_files,
successful_files=successful_files,
failed_files=failed_files,
execution_id=workflow_execution.id
execution_id=workflow_execution.id,
)

return workflow_execution
Expand Down

0 comments on commit f940d33

Please sign in to comment.