Ignore test requiring GPU

Ceyron · Sep 5, 2024 · 1b2f310 · 1b2f310
1 parent 0e5934d
commit 1b2f310
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/test_viz.py b/tests/test_viz.py
@@ -28,8 +28,10 @@ def test_plot_state_2d():
     plt.close(fig)
 
 
-def test_plot_state_3d():
-    state = jax.random.normal(jax.random.PRNGKey(0), (1, 32, 32, 32))
+# # Requires a GPU and therefore cannot easily be tested on GitHub Actions
 
-    fig = ex.viz.plot_state_3d(state)
-    plt.close(fig)
+# def test_plot_state_3d():
+#     state = jax.random.normal(jax.random.PRNGKey(0), (1, 32, 32, 32))
+
+#     fig = ex.viz.plot_state_3d(state)
+#     plt.close(fig)