Skip to content

Commit

Permalink
Fix formatting issue using pip black
Browse files Browse the repository at this point in the history
  • Loading branch information
dhruvsgarg committed Mar 23, 2024
1 parent d393ac7 commit c8b5608
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions rpc/service.py
Original file line number Diff line number Diff line change
Expand Up @@ -714,7 +714,7 @@ async def NotifyTaskCompletion(self, request, context):
message=f"Task with ID {request.task_id} "
f"not found in TaskGraph {request.application_id}.",
)

# Instead of completing & removing the task immediately, check
# if it is actually complete or will complete in the future

Expand All @@ -741,7 +741,7 @@ async def NotifyTaskCompletion(self, request, context):
# Else it will be dequeued at its actual task completion time
self._tasks_marked_for_completion.put(
TimedItem(actual_task_completion_time, matched_task)
)
)

# NOTE: task.finish() and run_scheduler() invocations are postponed
# until it is time for the task to be actually marked as complete.
Expand Down Expand Up @@ -830,7 +830,7 @@ async def PopTasksBasedOnTime(self):
popped_item.task.update_remaining_time(EventTime.zero())
popped_item.task.finish(
EventTime(round(current_time), EventTime.Unit.S)
)
)

# Run the scheduler since the Workload has changed.
await self.run_scheduler()
Expand Down

0 comments on commit c8b5608

Please sign in to comment.