Disable MPS

Rose-STL-Lab · Apr 29, 2024 · f6d9c97 · f6d9c97
1 parent 9583ae5
commit f6d9c97
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -37,6 +37,8 @@ jobs:
 
       - name: Run tests
         run: poetry run pytest tests/ --cov=torchts
+        env:
+          PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
 
       - name: Generate coverage report
         run: poetry run coverage xml