Skip to content

Commit

Permalink
remove plotting
Browse files Browse the repository at this point in the history
  • Loading branch information
MarleneBusch committed Mar 4, 2024
1 parent 3970da2 commit 6c301d2
Showing 1 changed file with 1 addition and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -182,8 +182,6 @@ def test_compute_bitmaps(environment_data: Dict[str, torch.Tensor]) -> None:
receiver_plane_x,
receiver_plane_y,
)
# plt.imshow(total_bitmap.T, origin="lower", cmap="jet")
# plt.show()

expected_path = (
pathlib.Path(ARTIST_ROOT)
Expand All @@ -192,6 +190,5 @@ def test_compute_bitmaps(environment_data: Dict[str, torch.Tensor]) -> None:
)

expected = torch.load(expected_path)
# plt.imshow(expected.T, origin="lower", cmap="jet")
# plt.show()

torch.testing.assert_close(total_bitmap, expected, atol=1.7e-5, rtol=2.6e-5)

0 comments on commit 6c301d2

Please sign in to comment.