From 116809096c2749dd420917fd2bbb40b9b0079a1b Mon Sep 17 00:00:00 2001 From: MohamedElgammal Date: Mon, 12 Aug 2024 20:04:47 -0400 Subject: [PATCH] fix lint runs --- vtr_flow/scripts/tuning_runs/control_runs.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vtr_flow/scripts/tuning_runs/control_runs.py b/vtr_flow/scripts/tuning_runs/control_runs.py index efe48621891..825b3b7609c 100755 --- a/vtr_flow/scripts/tuning_runs/control_runs.py +++ b/vtr_flow/scripts/tuning_runs/control_runs.py @@ -172,8 +172,9 @@ def generate_avg_seed_csv(full_res_csv_path, output_dir): col_to_keep.extend(list(PARAMS_DICT.keys())) col_to_keep.extend(PARSED_METRICS) df = df.drop( + # pylint: disable=no-member columns=[col for col in df.columns if col not in col_to_keep] - ) # pylint: disable=no-member + ) # Check if '--seed' column is present if "--seed" in df.columns: