envvars are strings

Author: stefdoerr

.github/workflows/CI.yml

@@ -77,7 +77,7 @@ jobs:
             pip -vv install .
           fi
         env:
-          WITH_CUDA: 0
+          WITH_CUDA: "0"
 
       - name: Lint with flake8
         run: |
@@ -89,7 +89,7 @@ jobs:
       - name: Run tests
         run: pytest -v -s --durations=10
         env:
-          WITH_CUDA: 0
+          WITH_CUDA: "0"
           SKIP_TORCH_COMPILE: ${{ runner.os == 'Windows' && 'true' || 'false' }}
           OMP_PREFIX: ${{ runner.os == 'macOS' && '/Users/runner/miniconda3/envs/test' || '' }}
           CPU_TRAIN: ${{ runner.os == 'macOS' && 'true' || 'false' }}

setup.py

@@ -27,7 +27,7 @@
     version = "0"
 
 # If WITH_CUDA is defined
-if os.environ.get("WITH_CUDA", 0) == 1:
+if os.environ.get("WITH_CUDA", "0") == "1":
     use_cuda = True
 else:
     use_cuda = torch.cuda._is_compiled()