Run unit tests on MacOS in CI (#83)

* Run unit tests on MacOS in CI Signed-off-by: Fabrice Normandin <[email protected]> * Remove specific cuda library reqs from 'gpu' extra Signed-off-by: Fabrice Normandin <[email protected]> * Add a skipif mark on test for MacOS in CI Signed-off-by: Fabrice Normandin <[email protected]> --------- Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Nov 7, 2024 · bc5723f · bc5723f
1 parent e8514f6
commit bc5723f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 44 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -57,7 +57,7 @@ jobs:
     strategy:
       max-parallel: 4
       matrix:
-        platform: [ubuntu-latest]
+        platform: [ubuntu-latest, macos-latest]
         python-version: ["3.10"]
     steps:
       - uses: actions/checkout@v4

diff --git a/project/main_test.py b/project/main_test.py
@@ -223,6 +223,10 @@ def test_example_experiment_defaults(experiment_config: Config) -> None:
     )
 
 
+@pytest.mark.skipif(
+    IN_GITHUB_CI and sys.platform == "darwin",
+    reason="TODO: Getting a 'MPS backend out of memory' error on the Github CI. ",
+)
 @pytest.mark.parametrize(
     command_line_overrides.__name__,
     [

diff --git a/pyproject.toml b/pyproject.toml
@@ -61,20 +61,7 @@ docs = [
     "mkdocs-macros-plugin>=1.0.5",
 ]
 gpu = [
-    "jax[cuda12]>=0.4.31",
-    "nvidia-cublas-cu12==12.1.3.1",
-    "nvidia-cuda-cupti-cu12==12.1.105",
-    "nvidia-cuda-nvcc-cu12==12.6.68",
-    "nvidia-cuda-nvrtc-cu12==12.1.105",
-    "nvidia-cuda-runtime-cu12==12.1.105",
-    "nvidia-cudnn-cu12==9.1.0.70",
-    "nvidia-cufft-cu12==11.0.2.54",
-    "nvidia-curand-cu12==10.3.2.106",
-    "nvidia-cusolver-cu12==11.4.5.107",
-    "nvidia-cusparse-cu12==12.1.0.106",
-    "nvidia-nccl-cu12==2.20.5",
-    "nvidia-nvjitlink-cu12==12.6.68",
-    "nvidia-nvtx-cu12==12.1.105",
+    "jax[cuda12]>=0.4.31"
 ]
 
 [build-system]

diff --git a/uv.lock b/uv.lock