diff --git a/.github/actions/setup-uv-project/action.yml b/.github/actions/setup-uv-project/action.yml index 7e23075a..bdaa935a 100644 --- a/.github/actions/setup-uv-project/action.yml +++ b/.github/actions/setup-uv-project/action.yml @@ -10,4 +10,4 @@ runs: enable-cache: true - shell: bash - run: uv sync --frozen --extra dev + run: uv sync --extra dev diff --git a/.github/workflows/installation.yaml b/.github/workflows/installation.yaml index 0f4fb8a6..3ba3184a 100644 --- a/.github/workflows/installation.yaml +++ b/.github/workflows/installation.yaml @@ -29,7 +29,3 @@ jobs: python-version: ${{ matrix.python-version }} - uses: ./.github/actions/setup-uv-project - - - name: Install dependencies - run: | - pip install -e . diff --git a/pyproject.toml b/pyproject.toml index d41dd4a8..119d3418 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -45,9 +45,20 @@ testpaths = [ "src/pruna/tests", ] -[tool.uv] -index-url = "https://pypi.org/simple/" -extra-index-url = ["https://prunaai.pythonanywhere.com/simple/"] +[[tool.uv.index]] +name = "pypi" +url = "https://pypi.org/simple" # regular PyPI + +[[tool.uv.index]] +name = "pruna_internal" +url = "https://prunaai.pythonanywhere.com/simple/" # Pruna Pythonanywhere +default = true # default = True makes this index the lowest prio + +[tool.uv.sources] +gptqmodel = [ + { index = "pruna_internal", marker = "sys_platform != 'darwin' or platform_machine != 'arm64'"}, + { index = "pypi", marker = "sys_platform == 'darwin' and platform_machine == 'arm64'"}, +] [project] name = "pruna" @@ -95,7 +106,7 @@ dependencies = [ "pynvml", "thop", "timm", - "bitsandbytes", + "bitsandbytes; sys_platform != 'darwin' or platform_machine != 'arm64'", "optimum-quanto>=0.2.5", "optimum", "ctranslate2==4.6.0",