Attempt to fix MPS out of memory error.

cboulay · cboulay · commit 2c092c60136d · 2025-08-22T12:57:38.000-04:00
diff --git a/.github/workflows/python-tests.yml b/.github/workflows/python-tests.yml
@@ -39,3 +39,5 @@ jobs:
 
       - name: Run tests
         run: uv run pytest tests
+        env:
+          PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
