fix fitness plots

src/blop/plotting.py

@@ -157,6 +157,8 @@ def _plot_objs_many_dofs(agent, axes=(0, 1), shading="nearest", cmap=DEFAULT_COL
 
     model_inputs = agent.dofs(active=True).transform(test_inputs)
 
+    print(agent.objectives)
+
     for obj_index, obj in enumerate(agent.objectives):
         targets = agent.train_targets()[obj.name].numpy()
 
@@ -192,7 +194,7 @@ def _plot_objs_many_dofs(agent, axes=(0, 1), shading="nearest", cmap=DEFAULT_COL
             #     cmap=cmap,
             #     norm=val_norm,
             # )
-            if obj.constraint is not None:
+            if obj.constraint is None:
                 fitness_ax = agent.obj_axes[obj_index, 1].pcolormesh(
                     test_x,
                     test_y,