Merge branch 'gabi/update_v1.1' into gabi/update_v1.1_a

Author: gabrieltseng

.github/workflows/publish.yml

@@ -24,11 +24,11 @@ jobs:
       - uses: actions/checkout@v5
         with:
           ref: ${{ github.event.inputs.tag || github.ref }}
-      
+
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
-          python-version: "3.12"
+          python-version: ".python-version"
 
       - name: Verify version matches tag
         run: |

.github/workflows/tests.yml

@@ -10,7 +10,7 @@ on:
 
 jobs:
   test:
-    name: Test with torch-cpu
+    name: Tests
     runs-on: ubuntu-latest
 
     steps:
@@ -28,11 +28,8 @@ jobs:
 
     - name: Install dependencies
       run: |
-        # Install with CPU-only torch (suitable for CI)
         uv sync --python 3.12 --extra dev
         source .venv/bin/activate
-        # Install the project itself
-        uv pip install -e .
 
     - name: Run tests (without weights)
       run: |

.gitignore

@@ -1,5 +1,5 @@
 __pycache__/
-**.egg-info
+**/*.egg-info/
 # we ignore the lockfile since
 # we have gpu and cpu installations.
 uv.lock

.python-version

@@ -0,0 +1 @@
+3.12

README.md

@@ -23,7 +23,7 @@ To install dependencies:
 ```
 git clone git@github.com:allenai/olmoearth_pretrain_minimal.git
 cd olmoearth_pretrain_minimal
-uv sync --python 3.12
+uv sync
 ```
 uv installs everything into a venv, so to keep using python commands you can activate uv's venv: `source .venv/bin/activate`. Otherwise, swap to `uv run python`.