change np.product to np.prod

neuromorphs · Jul 10, 2024 · 86ad1c4 · 86ad1c4
1 parent fe4ecf4
commit 86ad1c4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/test/test_tonic_utils.py b/test/test_tonic_utils.py
@@ -47,13 +47,13 @@ def test_pytorch_batch_collation_dense_tensor():
     assert batch.shape[2] == sensor_size[2]
 
 
-@pytest.mark.skipif(not platform == "linux", reason="Test doesn't finish on Windows.")
+@pytest.mark.skipif(platform == "win32", reason="Test doesn't finish on Windows.")
 def test_plotting():
     events, sensor_size = create_random_input()
     tonic.utils.plot_event_grid(events)
 
 
-@pytest.mark.skipif(not platform == "linux", reason="Test doesn't finish on Windows.")
+@pytest.mark.skipif(platform == "win32", reason="Test doesn't finish on Windows.")
 def test_animation():
     events, sensor_size = create_random_input()
 

diff --git a/tonic/utils.py b/tonic/utils.py
@@ -42,7 +42,7 @@ def plot_event_grid(
         sensor_size = (sensor_size_x, sensor_size_y, sensor_size_p)
 
         transform = transforms.ToFrame(
-            sensor_size=sensor_size, n_time_bins=np.product(axis_array)
+            sensor_size=sensor_size, n_time_bins=np.prod(axis_array)
         )
 
         frames = transform(events)