Skip to content

Commit

Permalink
made changes to results gathering
Browse files Browse the repository at this point in the history
  • Loading branch information
achew010 committed May 27, 2024
1 parent 04702cc commit 5de954f
Show file tree
Hide file tree
Showing 3 changed files with 80 additions and 82 deletions.
10 changes: 5 additions & 5 deletions scripts/benchmarks/display_bench_results.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,14 +6,14 @@
from scripts.benchmarks.benchmark import gather_report, DIR_SAMP_CONFIGS
from typing import List

def main(*directories: str, output_filename: str = "results.csv", remove_columns: List[str]):
def main(*directories: str, output_filename: str = "results.csv", remove_columns: List[str] = None):
"gather outputs from a list of directories and output to a csv"

df, constant = gather_report(*directories, raw=False)
# filter result columns to keep by the inverse of remove_columns
df = df[df.columns[~df.columns.isin(remove_columns)]]
# remove root dir from config filename
df["acceleration_framework_config_file"] = df["acceleration_framework_config_file"].str.replace(f"{DIR_SAMP_CONFIGS}", "sample-configurations")
if remove_columns:
df = df[df.columns[~df.columns.isin(remove_columns)]]

errors = []
try:
# remove error messages if any
Expand Down Expand Up @@ -50,7 +50,7 @@ def main(*directories: str, output_filename: str = "results.csv", remove_columns
)
parser.add_argument(
"--remove_columns",
nargs="+",
nargs="*",
help="list of columns to ignore from results.csv",
)

Expand Down
Loading

0 comments on commit 5de954f

Please sign in to comment.