Skip to content

Commit

Permalink
fix pre commit failure with black
Browse files Browse the repository at this point in the history
  • Loading branch information
jennmald committed Jan 2, 2025
1 parent a129687 commit b71262d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/blop/plotting.py
Original file line number Diff line number Diff line change
Expand Up @@ -181,7 +181,7 @@ def _plot_objs_many_dofs(agent, axes=(0, 1), shading="nearest", cmap=DEFAULT_COL

test_constraint = None
if obj.constraint is None:
#test_constraint = obj.constraint_probability(model_inputs).detach().squeeze().numpy()
# test_constraint = obj.constraint_probability(model_inputs).detach().squeeze().numpy()
test_constraint = agent.constraint(model_inputs).squeeze().numpy()

if gridded:
Expand Down Expand Up @@ -218,7 +218,7 @@ def _plot_objs_many_dofs(agent, axes=(0, 1), shading="nearest", cmap=DEFAULT_COL
test_constraint,
shading=shading,
cmap=cmap,
#norm=mpl.colors.LogNorm(),
# norm=mpl.colors.LogNorm(),
)

else:
Expand Down

0 comments on commit b71262d

Please sign in to comment.