From c34b55c609d31e0a178853f0a2e9a195f1690b96 Mon Sep 17 00:00:00 2001 From: Dhruv Garg Date: Tue, 27 Feb 2024 19:36:23 -0500 Subject: [PATCH] fixes using flake8 linting --- rpc/service.py | 4 ++-- rpc/tpch_utils.py | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/rpc/service.py b/rpc/service.py index 315fcb28..8934d80c 100644 --- a/rpc/service.py +++ b/rpc/service.py @@ -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, ) diff --git a/rpc/tpch_utils.py b/rpc/tpch_utils.py index c8ff9b6e..ebc4e3cd 100644 --- a/rpc/tpch_utils.py +++ b/rpc/tpch_utils.py @@ -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 = { @@ -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