forgot to run precommit

NSLS-II · Feb 3, 2025 · 49752b5 · 49752b5
1 parent 3fa3192
commit 49752b5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git 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: