Enable sim_no_compile test in CI.

PiperOrigin-RevId: 629692455
google-deepmind · May 1, 2024 · 5befd7f · 5befd7f
1 parent e41fd1b
commit 5befd7f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -98,6 +98,8 @@ jobs:
           -vv -n auto \
           --shard-id=$((${{ matrix.shard-id }} - 1)) --num-shards=${{ env.PYTEST_NUM_SHARDS }}
 
-    # # TODO(b/326579500): pass env variable and run under pytest
-    # - name: "Run torax/tests/sim_no_compile.py"
-    #   run: TORAX_COMPILATION_ENABLED=False python torax/tests/sim_no_compile.py
+    # One test requires an extra environment variable, so we run it separately.
+    - name: Run sim_no_compile
+      if: ${{ matrix.shard-id == 1 }}
+      run: |
+        TORAX_COMPILATION_ENABLED=False pytest -vv -n auto torax/tests/sim_no_compile.py