Commit 941437ed authored by Connor Hainje's avatar Connor Hainje
Browse files

Formatting, add kwargs to some functions

parent 66fd4d95
......@@ -194,7 +194,9 @@ def _subplot_2D(
vmax = len(const.PARTICLES) - 0.5
else:
vmin, vmax = vlims
im = ax.imshow(idx, cmap=cmap, vmin=vmin, vmax=vmax, aspect="auto", origin="lower")
im = ax.imshow(
idx, cmap=cmap, vmin=vmin, vmax=vmax, aspect="auto", origin="lower"
)
connected_region_borders(ax, idx)
elif particle in const.PARTICLES:
......@@ -208,7 +210,9 @@ def _subplot_2D(
else:
vmin, vmax = vlims
im = ax.imshow(val, cmap=cmap, vmin=vmin, vmax=vmax, aspect="auto", origin="lower")
im = ax.imshow(
val, cmap=cmap, vmin=vmin, vmax=vmax, aspect="auto", origin="lower"
)
else:
raise ValueError("particle not understood")
......@@ -269,10 +273,10 @@ def _plot_2D(
title=None,
legend_or_cbar=None,
legend_cb_label="",
cell_fontsize=10,
**kwargs,
):
fig, ax = plt.subplots(figsize=figsize)
im = subplot_fn(ax, df, particle=particle, cell_fontsize=cell_fontsize)
im = subplot_fn(ax, df, particle=particle, **kwargs)
fig.tight_layout()
imshow_2D_axis_labels(ax)
better_xlabel(fig, title, loc="top")
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment