Skip to content

Commit

Permalink
fix gather
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed Nov 15, 2023
1 parent 1a05535 commit d403046
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 7 deletions.
2 changes: 0 additions & 2 deletions optimum_benchmark/aggregators/gather.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,8 +27,6 @@ def gather(root_folders: List[Path]) -> pd.DataFrame:

if (len(results_dfs) == 0) or (len(configs_dfs) == 0):
raise ValueError(f"No results found in {root_folder}")
elif len(results_dfs) != len(configs_dfs):
raise ValueError(f"Results and configs do not match in {root_folder}")

# Merge inference and config dataframes
full_dfs = {}
Expand Down
13 changes: 8 additions & 5 deletions optimum_benchmark/report.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,5 @@
import sys

from .aggregators.display import display_cli
from .aggregators.gather import gather_cli
from .aggregators.plot import plot_cli
from .aggregators.summarize import summarize_cli

HELP = """
Usage: optimum-report <action> <options>
Actions:
Expand All @@ -24,12 +19,20 @@ def main():
sys.argv = sys.argv[1:]

if action == "gather":
from .aggregators.gather import gather_cli

gather_cli()
elif action == "display":
from .aggregators.display import display_cli

display_cli()
elif action == "summarize":
from .aggregators.summarize import summarize_cli

summarize_cli()
elif action == "plot":
from .aggregators.plot import plot_cli

plot_cli()
elif action in ["-h", "--help"]:
print(HELP)
Expand Down

0 comments on commit d403046

Please sign in to comment.