[BugFix] MacOS testing

.github/workflows/tests.yml

@@ -17,6 +17,10 @@ jobs:
     - name: Check out repository
       uses: actions/checkout@v3
 
+    - name: Set environment variable for macOS
+      if: runner.os == 'macOS'
+      run: echo "MAC_OS_GITHUB_RUNNER=1" >> $GITHUB_ENV
+
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/setup-python@v4
       with:

tests/training.py

@@ -1,8 +1,16 @@
+import os
+
 from rl4co.utils.trainer import RL4COTrainer
 
 from routefinder.envs.mtvrp import MTVRPEnv
 from routefinder.models import RouteFinderBase, RouteFinderPolicy
 
+# Get env variable MAC_OS_GITHUB_RUNNER and force CPU in that case
+if "MAC_OS_GITHUB_RUNNER" in os.environ:
+    accelerator = "cpu"
+else:
+    accelerator = "auto"
+
 
 def test_training():
     env = MTVRPEnv(generator_params={"num_loc": 10, "variant_preset": "all"})
@@ -20,7 +28,7 @@ def test_training():
         max_epochs=1,
         gradient_clip_val=None,
         devices=1,
-        accelerator="auto",
+        accelerator=accelerator,
     )
     trainer.fit(model)
     trainer.test(model)