use PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0 for macos test

SBU-BMI · Nov 11, 2023 · 3738fdf · 3738fdf
1 parent b4811b4
commit 3738fdf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -101,7 +101,7 @@ jobs:
           mkdir slides && cd slides
           wget -q https://openslide.cs.cmu.edu/download/openslide-testdata/Aperio/JP2K-33003-1.svs
           cd ..
-          wsinfer run --wsi-dir slides/ --results-dir results/ --model breast-tumor-resnet34.tcga-brca
+          PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.0 wsinfer run --wsi-dir slides/ --results-dir results/ --model breast-tumor-resnet34.tcga-brca
           test -f results/run_metadata_*.json
           test -f results/patches/JP2K-33003-1.h5
           test -f results/model-outputs-csv/JP2K-33003-1.csv