feat: switch pruna from poetry to uv

.github/workflows/cpu_tests.yaml

@@ -22,32 +22,22 @@ jobs:
 
     steps:
       - name: Checkout code
-        uses: actions/checkout@v3
+        uses: actions/checkout@v4
 
       - name: Set up Python
-        uses: actions/setup-python@v4
+        uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Install Poetry
-        uses: abatilo/actions-poetry@v2
-
-      - name: Update Poetry lock file
-        run: poetry lock
-
-      - name: Cache Poetry virtualenv
-        uses: actions/cache@v3
+      - uses: astral-sh/setup-uv@v6
         with:
-          path: ~/.cache/pypoetry/virtualenvs
-          key: poetry-${{ runner.os }}-${{ matrix.python-version }}-${{ hashFiles('**/poetry.lock') }}
-          restore-keys: |
-            poetry-${{ runner.os }}-${{ matrix.python-version }}
-
-      - name: Install dependencies using Poetry
-        run: poetry install --extras tests
+          enable-cache: true
+      - name: Install dependencies
+        run: |
+          uv sync --extra tests
 
       - name: Run tests
         run: |
-          poetry run pytest -m "cpu and not slow and not high_cpu"
+          uv run pytest -m "cpu and not slow and not high_cpu"
 
 

.github/workflows/documentation.yaml

@@ -18,32 +18,28 @@ jobs:
       matrix:
         python-version: ["3.10"]
 
-
     steps:
       - name: Checkout code
-        uses: actions/checkout@v3
+        uses: actions/checkout@v4
 
       - name: Set up Python
-        uses: actions/setup-python@v3
+        uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Install Poetry
-        run: |
-          curl -sSL https://install.python-poetry.org | python3 -
-          echo "${HOME}/.local/bin" >> $GITHUB_PATH
-
-      - name: Configure Poetry to not create virtual environments
-        run: poetry config virtualenvs.create false
+      - uses: astral-sh/setup-uv@v6
+        with:
+          enable-cache: true
 
       - name: Install dependencies
-        run: pip install -e .[cpu]
+        run: |
+          uv pip install -e .[cpu]
 
       - name: Remove .so files
         run: find . -name "*.so" -delete
 
       - name: Run generation script
-        run: python docs/utils/gen_docs.py
+        run: uv run docs/utils/gen_docs.py
 
       - name: Verify generated file exists
         run: |

.github/workflows/installation.yaml

@@ -19,27 +19,19 @@ jobs:
         os: [ubuntu-latest, macos-latest, windows-latest]
         python-version: ["3.9", "3.10", "3.11", "3.12"]
 
-
     steps:
       - name: Checkout code
-        uses: actions/checkout@v3
+        uses: actions/checkout@v4
 
       - name: Set up Python
-        uses: actions/setup-python@v3
+        uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Install Poetry on Windows
-        if: runner.os == 'Windows'
-        run: |
-          powershell -Command "(Invoke-WebRequest -Uri https://install.python-poetry.org -UseBasicParsing).Content | python - --version 1.8.5"
-          echo "$(python -m site --user-base)\Scripts" >> $env:GITHUB_PATH
-
-      - name: Install Poetry on Unix
-        if: runner.os != 'Windows'
-        run: |
-          curl -sSL https://install.python-poetry.org | POETRY_VERSION=1.8.5 python3 -
-          echo "$(python3 -m site --user-base)/bin" >> $GITHUB_PATH
+      - uses: astral-sh/setup-uv@v6
+        with:
+          enable-cache: true
 
       - name: Install dependencies
-        run: pip install -e .
+        run: |
+          uv pip install -e .

.github/workflows/linting.yaml

@@ -18,41 +18,27 @@ jobs:
         python-version: ["3.10"]
 
     steps:
-      - name: Checkout code
-        uses: actions/checkout@v3
+      - uses: actions/checkout@v4
 
-      - name: Set up Python
-        uses: actions/setup-python@v4
+      - uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Install Poetry
-        uses: abatilo/actions-poetry@v2
-
-      - name: Update Poetry lock file
-        run: poetry lock
-
-      - name: Cache Poetry virtualenv
-        uses: actions/cache@v3
+      - uses: astral-sh/setup-uv@v6
         with:
-          path: ~/.cache/pypoetry/virtualenvs
-          key: poetry-${{ runner.os }}-${{ matrix.python-version }}-${{ hashFiles('**/poetry.lock') }}
-          restore-keys: |
-            poetry-${{ runner.os }}-${{ matrix.python-version }}
-
-      - name: Install dependencies using Poetry
-        run: poetry install --extras tests
+          enable-cache: true
 
-      - name: Remove .so files
-        run: find . -name "*.so" -delete
+      - name: Install dependencies
+        run: |
+          uv sync --extra tests
 
       - name: Run ruff on Pruna code
         uses: astral-sh/ruff-action@v3
         with:
           version: "latest"
 
       - name: Run mypy on Pruna code
-        run: poetry run mypy --show-traceback src/pruna
+        run: uv run mypy --show-traceback src/pruna
 
       - name: Run docstring checks with pytest
-        run: poetry run pytest -m "style"
+        run: uv run pytest -m "style"

.github/workflows/package_build.yaml

@@ -16,17 +16,18 @@ jobs:
 
     steps:
       - name: Checkout repository
-        uses: actions/checkout@v3
+        uses: actions/checkout@v4
 
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v3
+        uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Install uv
-        uses: astral-sh/setup-uv@v3
+      - uses: astral-sh/setup-uv@v6
+        with:
+          enable-cache: true
 
-      - name: Build Wheel
+      - name: Build
         run: |
           uv build
 
@@ -40,4 +41,5 @@ jobs:
         uses: actions/upload-artifact@v4
         with:
           name: ${{ steps.get_wheel.outputs.wheel_name }}
-          path: dist/*.whl
+          path: dist/*.whl
+          retention-days: 7

.gitignore

@@ -99,13 +99,6 @@ ipython_config.py
 #   install all needed dependencies.
 #Pipfile.lock
 
-# poetry
-#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
-#   This is especially recommended for binary packages to ensure reproducibility, and is more
-#   commonly ignored for libraries.
-#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
-#poetry.lock
-
 # pdm
 #   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
 #pdm.lock

pyproject.toml

@@ -5,7 +5,7 @@ line-length = 121
 [tool.ruff.lint]
 ignore = ["ANN002", "ANN003", "ANN401", "C901", "D100", "D104", "D401", "D406", "C408"]
 select = ["A", "C", "CPY", "D", "E", "ERA", "F", "FIX", "I", "N", "SIM", "T20", "W", "FA"]
-preview = true  # enable preview features for copyright checking
+preview = true # enable preview features for copyright checking
 
 [tool.ruff.lint.per-file-ignores]
 "*.ipynb" = ["ERA", "CPY", "T"]  # commented out code is intended for tutorials
@@ -31,7 +31,7 @@ description = "Smash your AI models"
 authors = [
     {name = "Pruna AI", email = "hello@pruna.ai"}
 ]
-license = {text = "Apache-2.0"}
+license = {file = "LICENSE"}
 readme = "README.md"
 requires-python = ">=3.9,<3.13"
 keywords = ["AI", "machine learning", "model optimization", "pruning"]
@@ -107,7 +107,8 @@ gptq = [
     "threadpoolctl",
     "device-smi",
     "random-word",
-    "gptqmodel==4.0.0.dev0+cu126torch2.7",
+    "gptqmodel==4.0.0.dev0+cu126torch2.7; sys_platform != 'darwin' or platform_machine != 'arm64'",
+    "gptqmodel; sys_platform == 'darwin' and platform_machine == 'arm64'",
     "logbar",
     "tokenicer",
     "threadpoolctl",
@@ -159,4 +160,8 @@ include = ["**/*.so"]
 exclude = [
     "tests*",
     "docs*",
-]
+]
+
+[[tool.uv.index]]
+name = "pruna"
+url = "https://prunaai.pythonanywhere.com/"