Add osx-arm64 platform to conda-lock.yml file and GitHub Actions CI

.github/workflows/test.yml

@@ -19,7 +19,7 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ["3.11"]
-        os: [ubuntu-22.04, macos-12]
+        os: [ubuntu-22.04, macos-12, macos-14]
     defaults:
       run:
         shell: bash -l {0}
@@ -41,3 +41,5 @@ jobs:
         run: |
             micromamba install python=${{ matrix.python-version }} pytest
             python -m pytest --verbose src/tests/
+        env:
+          PYTORCH_MPS_PREFER_METAL: 0  # disable MPS which runs out of memory on macos-14 runner

.pre-commit-config.yaml

@@ -6,6 +6,7 @@ repos:
   hooks:
     - id: check-added-large-files
       args: [ '--maxkb=512', '--enforce-all' ]
+      exclude: conda-lock.yml
     - id: check-yaml
     - id: end-of-file-fixer
     - id: trailing-whitespace

README.md

@@ -23,15 +23,19 @@ Launch into a [JupyterLab](https://jupyterlab.readthedocs.io) environment on
 To help out with development, start by cloning this [repo-url](/../../)
 
     git clone <repo-url>
+    cd model
 
 Then we recommend [using mamba](https://mamba.readthedocs.io/en/latest/installation.html)
-to install the dependencies.
-A virtual environment will also be created with Python and
+to install the dependencies. A virtual environment will also be created with Python and
 [JupyterLab](https://github.com/jupyterlab/jupyterlab) installed.
 
-    cd model
     mamba env create --file environment.yml
 
+> [!NOTE]
+> The command above will only work for Linux devices with CUDA GPUs. For installation
+> on macOS devices (either Intel or ARM chips), follow the 'Advanced' section in
+> https://clay-foundation.github.io/model/installation.html#advanced
+
 Activate the virtual environment first.
 
     mamba activate claymodel
@@ -40,27 +44,6 @@ Finally, double-check that the libraries have been installed.
 
     mamba list
 
-### Advanced
-
-This is for those who want full reproducibility of the virtual environment.
-Create a virtual environment with just Python and conda-lock installed first.
-
-    mamba create --name claymodel python=3.11 conda-lock=2.5.1
-    mamba activate claymodel
-
-Generate a unified [`conda-lock.yml`](https://github.com/conda/conda-lock) file
-based on the dependency specification in `environment.yml`. Use only when
-creating a new `conda-lock.yml` file or refreshing an existing one.
-
-    conda-lock lock --mamba --file environment.yml --platform linux-64 --with-cuda=12.0
-
-Installing/Updating a virtual environment from a lockile. Use this to sync your
-dependencies to the exact versions in the `conda-lock.yml` file.
-
-    conda-lock install --mamba --name claymodel conda-lock.yml
-
-See also https://conda.github.io/conda-lock/output/#unified-lockfile for more
-usage details.
 
 ## Usage