Skip to content

Commit

Permalink
fixes using flake8 linting
Browse files Browse the repository at this point in the history
  • Loading branch information
dhruvsgarg committed Feb 28, 2024
1 parent c99b12d commit c34b55c
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions rpc/service.py
Original file line number Diff line number Diff line change
Expand Up @@ -396,8 +396,8 @@ async def RegisterTaskGraph(self, request, context):
)
return erdos_scheduler_pb2.RegisterTaskGraphResponse(
success=False,
message=f"TPCH application ID {request.id} with name {request.name} "
f"couldn't be registered. DAG structure mismatch!",
message=f"TPCH application ID {request.id} with name {request.name}"
f" couldn't be registered. DAG structure mismatch!",
num_executors=0,
)

Expand Down
8 changes: 4 additions & 4 deletions rpc/tpch_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,9 +80,9 @@ def get_all_stage_info_for_query(query_num):
# drag nearest neighbor first wave duration to empty spots
pre_process_task_duration(task_duration)
rough_duration = np.mean(
[i for l in task_duration["first_wave"].values() for i in l]
+ [i for l in task_duration["rest_wave"].values() for i in l]
+ [i for l in task_duration["fresh_durations"].values() for i in l]
[i for t in task_duration["first_wave"].values() for i in t]
+ [i for t in task_duration["rest_wave"].values() for i in t]
+ [i for t in task_duration["fresh_durations"].values() for i in t]
)

curr_stage = {
Expand Down Expand Up @@ -153,7 +153,7 @@ def are_structurally_same(graph1, graph2):

return True, reversed_mapping

print(f"DAG structure mismatch! No mapping could be found")
print("DAG structure mismatch! No mapping could be found")
return False, None


Expand Down

0 comments on commit c34b55c

Please sign in to comment.