Skip to content

Commit d63d2c2

Browse files
author
MohamedElgammal
committed
try fix lint
1 parent 2431ec2 commit d63d2c2

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

vtr_flow/scripts/tuning_runs/control_runs.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import os
77
import sys
88
import csv
9-
from typing import Optional
109
import pandas as pd
1110
import numpy as np
1211
from scipy import stats
@@ -165,14 +164,16 @@ def parse_script_params(script_params):
165164

166165
def generate_avg_seed_csv(full_res_csv_path, output_dir):
167166
"""Generate the average results over the seeds"""
168-
df: Optional[pd.DataFrame] = pd.read_csv(full_res_csv_path)
167+
df = pd.read_csv(full_res_csv_path)
169168
assert isinstance(df, pd.DataFrame)
170169

171170
if KEEP_METRICS_ONLY:
172171
col_to_keep = ["circuit", "arch"]
173172
col_to_keep.extend(list(PARAMS_DICT.keys()))
174173
col_to_keep.extend(PARSED_METRICS)
175-
df = df.drop(columns=[col for col in df.columns if col not in col_to_keep])
174+
df = df.drop(
175+
columns=[col for col in df.columns if col not in col_to_keep]
176+
) # pylint: disable=no-member
176177

177178
# Check if '--seed' column is present
178179
if "--seed" in df.columns:
@@ -204,6 +205,7 @@ def generate_geomean_res_csv(full_res_csv_path, output_dir):
204205
param_columns = [key for key in PARAMS_DICT if key != "--seed"]
205206
non_param_columns = [col for col in df.columns if col not in param_columns]
206207

208+
# pylint: disable=no-member
207209
geomean_df = (
208210
df.groupby(param_columns)
209211
.agg(

0 commit comments

Comments
 (0)