From 49752b5d14b00e8ed7afbb277cf06ef1f3563e41 Mon Sep 17 00:00:00 2001 From: jennmald Date: Mon, 3 Feb 2025 13:55:10 -0500 Subject: [PATCH] forgot to run precommit --- src/blop/plotting.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/blop/plotting.py b/src/blop/plotting.py index e281757..2ac15ce 100644 --- a/src/blop/plotting.py +++ b/src/blop/plotting.py @@ -175,8 +175,10 @@ def _plot_objs_many_dofs(agent, axes=(0, 1), shading="nearest", cmap=DEFAULT_COL # mask for nan values, uses unfilled o marker mask = np.isnan(values) - val_ax = agent.obj_axes[obj_index, 0].scatter(x_values[~mask], y_values[~mask], c=values[~mask], s=size, norm=val_norm, cmap=cmap) - agent.obj_axes[obj_index, 0].scatter(x_values[mask], y_values[mask], marker="o", ec='k', fc='w', s=size) + val_ax = agent.obj_axes[obj_index, 0].scatter( + x_values[~mask], y_values[~mask], c=values[~mask], s=size, norm=val_norm, cmap=cmap + ) + agent.obj_axes[obj_index, 0].scatter(x_values[mask], y_values[mask], marker="o", ec="k", fc="w", s=size) # mean and sigma will have shape (*input_shape,) test_posterior = obj.model.posterior(model_inputs) @@ -269,7 +271,7 @@ def _plot_objs_many_dofs(agent, axes=(0, 1), shading="nearest", cmap=DEFAULT_COL norm=mpl.colors.LogNorm(), ) - val_cbar = agent.obj_fig.colorbar(val_ax, ax=agent.obj_axes[obj_index,0], location="bottom", aspect=32, shrink=0.8) + val_cbar = agent.obj_fig.colorbar(val_ax, ax=agent.obj_axes[obj_index, 0], location="bottom", aspect=32, shrink=0.8) val_cbar.set_label(f"{obj.units or ''}") if obj.constraint is None: