switch from pixi to uv

.github/workflows/main.yml

@@ -13,13 +13,11 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: prefix-dev/setup-pixi@v0.8.1
+      - uses: astral-sh/setup-uv@v3
         with:
-          pixi-version: v0.27.1
-          frozen: true
-          cache: true
-          environments: dev
-          activate-environment: dev
+          enable-cache: true
+          cache-dependency-glob: "uv.lock"
+      - run: uv sync
       - run: ruff check
         if: success() || failure()
       - run: ruff format --check
@@ -31,13 +29,11 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: prefix-dev/setup-pixi@v0.8.1
+      - uses: astral-sh/setup-uv@v3
         with:
-          pixi-version: v0.27.1
-          frozen: true
-          cache: true
-          environments: extra
-          activate-environment: extra
+          enable-cache: true
+          cache-dependency-glob: "uv.lock"
+      - run: uv sync
       - run: pyright
         if: success() || failure()
 
@@ -72,18 +68,11 @@ jobs:
 
     steps:
       - uses: actions/checkout@v4
-
-      - name: Inject (python / pytorch) test deps into pixi.toml / pyproject.toml
-        run: |
-          sed -i 's/python = "3.9"/python = "${{ matrix.python }}.*"/' pixi.toml
-          sed -i 's/torch>=2.0.0/torch~=${{ matrix.pytorch }}/' pyproject.toml
-
-      - uses: prefix-dev/setup-pixi@v0.8.1
+      - name: Override pytorch version
+        run: sed -i '/\[tool\.uv\]/a override-dependencies = \["torch==${{ matrix.pytorch }}"\]' pyproject.toml
+      - uses: astral-sh/setup-uv@v3
         with:
-          pixi-version: v0.27.1
-          locked: false
-          frozen: false
-          cache: false
-          environments: default
-          activate-environment: default
+          enable-cache: true
+          cache-dependency-glob: "uv.lock"
+      - run: uv sync --python ${{ matrix.python }}
       - run: pytest tests/test_ci.py

.github/workflows/release.yml

@@ -9,15 +9,10 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: prefix-dev/setup-pixi@v0.8.1
+      - uses: astral-sh/setup-uv@v3
         with:
-          pixi-version: v0.27.1
-          frozen: true
-          cache: true
-          environments: dev
-          activate-environment: dev
-      - run: python -m build
-      - run: twine upload dist/*
-        env:
-          TWINE_USERNAME: __token__
-          TWINE_PASSWORD: ${{ secrets.PYPI_TOKEN }}
+          enable-cache: true
+          cache-dependency-glob: "uv.lock"
+      - run: uv sync
+      - run: uv build
+      - run: uv publish

.python-version

@@ -0,0 +1 @@
+3.9.20

pyproject.toml

@@ -21,18 +21,25 @@ classifiers = [
 ]
 requires-python = ">=3.9"
 dependencies = [
-    "cloudpickle>=3.0.0",
-    "fabric>=3.0.0",
-    "torch>=2.0.0",
+    "cloudpickle>=3.0",
+    "fabric>=3.2",
+    "torch>=2.0,<2.5",
     # torch.distributed depends on numpy
-    "numpy<2",  # numpy>=2 requires torch>=2.3
-    "numpy>=1.26.0; python_version == '3.12'",
+    "numpy>=1.20",  # numpy>=2 requires torch>=2.3
 ]
 
 [project.urls]
 Repository = "https://github.com/apoorvkh/torchrunx.git"
 Documentation = "https://torchrunx.readthedocs.io"
 
+[tool.uv]
+constraint-dependencies = ["torch>=2.1.1"]  # https://github.com/pytorch/pytorch/issues/100974
+dev-dependencies = ["ruff==0.7.0", "pyright==1.1.385", "pytest==8.3.3"]
+environments = ["sys_platform == 'linux'"]
+managed = true
+python-preference = "only-managed"
+resolution = "lowest-direct"
+
 ## Development tools
 
 [tool.ruff]