Merge branch 'dev' into hyperviscous-kdv

Ceyron · Sep 5, 2024 · ecce3ec · ecce3ec
2 parents 4f5e085 + 1b2f310
commit ecce3ec
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
@@ -7,7 +7,7 @@ jobs:
   run-test:
     strategy:
       matrix:
-        python-version: [ 3.10, 3.11, 3.12 ]
+        python-version: [ "3.10", "3.11", "3.12" ]
         os: [ ubuntu-latest ]
       fail-fast: false
     runs-on: ${{ matrix.os }}

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)