Skip to content

Commit

Permalink
now can skip plotting of methods also
Browse files Browse the repository at this point in the history
  • Loading branch information
FilippoAiraldi committed Dec 27, 2023
1 parent f76df14 commit 148ef53
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions benchmarking/plot.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ def load_data(csv_filename: str, exclude: list[str]) -> pd.DataFrame:
dtype={"problem": pd.StringDtype(), "method": pd.StringDtype()},
converters={s: converter for s in ["stage-reward", "best-so-far", "time"]},
)
df = df[~df["method"].isin(exclude)]
df = df[~(df["method"].isin(exclude) | df["problem"].isin(exclude))]

# manually sort problems alphabetically but methods in a custom order
df.sort_values(
Expand Down Expand Up @@ -411,7 +411,7 @@ def summarize(df: pd.DataFrame, tabletitle: Optional[str]) -> None:
type=str,
nargs="+",
default=[],
help="List of methods to exclude from the visualization.",
help="List of methods and/or benchmarks to exclude from the visualization.",
)
parser.add_argument(
"--no-plot",
Expand Down

0 comments on commit 148ef53

Please sign in to comment.