Skip to content

docs: improve benchmark plot readability #32

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
Ramlaoui wants to merge 1 commit into from
Closed

docs: improve benchmark plot readability #32

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 9 additions & 9 deletions atompack-py/benchmarks/plot_benchmark_story.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -183,8 +183,8 @@ def _import_plotting():
"axes.spines.right": False,
"axes.spines.left": False,
"axes.spines.bottom": False,
"xtick.color": MUTED,
"ytick.color": MUTED,
"xtick.color": TEXT,
"ytick.color": TEXT,
"xtick.labelsize": 10.5,
"ytick.labelsize": 10.5,
"axes.labelsize": 12.5,
Expand Down Expand Up @@ -221,7 +221,7 @@ def _errorbar_style(backend: str) -> dict[str, object]:
}


EXCLUDED_BACKENDS: set[str] = set()
EXCLUDED_BACKENDS: set[str] = {"lmdb_soa"}
WRITE_EXCLUDED_BACKENDS = {"atompack_ase_batch"}


Expand Down Expand Up @@ -322,7 +322,7 @@ def _style_axes(ax, *, title: str, xlabel: str, ylabel: str = "", log_x: bool =
def _add_header(fig, *, title: str, subtitle: str = "") -> None:
fig.text(0.5, 0.945, title, ha="center", va="center", fontsize=22, fontweight="semibold", color=TEXT)
if subtitle:
fig.text(0.5, 0.912, subtitle, ha="center", va="center", fontsize=11, fontweight="regular", color=MUTED)
fig.text(0.5, 0.912, subtitle, ha="center", va="center", fontsize=11, fontweight="regular", color=TEXT)


def _add_panel_note(ax, text: str) -> None:
Expand All @@ -335,15 +335,15 @@ def _add_panel_note(ax, text: str) -> None:
va="bottom",
fontsize=10,
fontweight="medium",
color=MUTED,
color=TEXT,
clip_on=False,
)


def _add_footer(fig) -> None:
fig.text(
0.5, 0.005, "atompack benchmark suite",
ha="center", va="bottom", fontsize=8, color=MUTED, style="italic",
ha="center", va="bottom", fontsize=8, color=TEXT, style="italic",
)


Expand Down Expand Up @@ -388,7 +388,7 @@ def _plot_ranked_bars(
ax.set_yticklabels([_backend_label(row.backend) for row in ranked])
for tick, row in zip(ax.get_yticklabels(), ranked):
if row.backend == "atompack":
tick.set_fontweight("medium")
tick.set_fontweight("semibold")
tick.set_color(TEXT)
ax.invert_yaxis()
_style_axes(ax, title=title, xlabel="atoms/s" if show_xlabel else "", log_x=True)
Expand Down Expand Up @@ -419,7 +419,7 @@ def _plot_ranked_bars(
va="center",
fontsize=11,
fontweight="semibold" if row.backend == "atompack" else "medium",
color=_backend_color(row.backend),
color=TEXT,
)


Expand Down Expand Up @@ -559,7 +559,7 @@ def _plot_storage_ratio_lines(ax, rows: list[BenchmarkRow], *, title: str) -> tu
ha="left",
va="bottom",
fontsize=10,
color=MUTED,
color=TEXT,
)
ax.set_xlim(0.78, max(all_values) * 1.08)

Expand Down
Loading