File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
vtr_flow/scripts/tuning_runs Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 66import os
77import sys
88import csv
9- from typing import Optional
109import pandas as pd
1110import numpy as np
1211from scipy import stats
@@ -165,14 +164,16 @@ def parse_script_params(script_params):
165164
166165def 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 (
You can’t perform that action at this time.
0 commit comments