From 1b7e71a64076dbc43bc4fcb6395b8799441d9f41 Mon Sep 17 00:00:00 2001
From: Mat Sz <contact@matsz.dev>
Date: Sun, 21 Apr 2024 20:30:46 +0200
Subject: [PATCH] fix uv arguments

---
 .github/workflows/build.yml | 10 ++++++++--
 requirements.cpu.txt        |  1 -
 requirements.directml.txt   |  1 -
 requirements.rocm.txt       |  1 -
 4 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index c6d95ff..81dbfc6 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -13,16 +13,19 @@ jobs:
         build:
           - arch: 'x64'
             type: 'cuda'
+            uv_args: ''
           - arch: 'x64'
             type: 'cpu'
+            uv_args: '--prerelease=allow'
           - arch: 'x64'
             type: 'rocm'
+            uv_args: '--prerelease=allow'
     runs-on: ubuntu-latest
     env:
       artifact_name: linux-${{ matrix.build.arch }}-${{ matrix.build.type }}.tar.zst
       python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20240415/cpython-3.11.9+20240415-x86_64_v3-unknown-linux-gnu-install_only.tar.gz'
       uv_url: 'https://github.com/astral-sh/uv/releases/download/0.1.35/uv-x86_64-unknown-linux-gnu.tar.gz'
-      requirements: requirements.${{ matrix.build.type }}.txt
+      requirements: ${{ matrix.build.uv_args }} requirements.${{ matrix.build.type }}.txt
     steps:
       - uses: actions/checkout@v4
 
@@ -78,16 +81,19 @@ jobs:
         build:
           - arch: 'x64'
             type: 'cuda'
+            uv_args: ''
           - arch: 'x64'
             type: 'directml'
+            uv_args: '--prerelease=allow'
           - arch: 'x64'
             type: 'cpu'
+            uv_args: '--prerelease=allow'
     runs-on: windows-latest
     env:
       artifact_name: win32-${{ matrix.build.arch }}-${{ matrix.build.type }}.tar.zst
       python_url: 'https://github.com/indygreg/python-build-standalone/releases/download/20240415/cpython-3.11.9+20240415-x86_64-pc-windows-msvc-install_only.tar.gz'
       uv_url: 'https://github.com/astral-sh/uv/releases/download/0.1.35/uv-x86_64-pc-windows-msvc.zip'
-      requirements: requirements.${{ matrix.build.type }}.txt
+      requirements: ${{ matrix.build.uv_args }} requirements.${{ matrix.build.type }}.txt
     steps:
       - uses: actions/checkout@v4
 
diff --git a/requirements.cpu.txt b/requirements.cpu.txt
index f361b1d..d775cc7 100644
--- a/requirements.cpu.txt
+++ b/requirements.cpu.txt
@@ -1,4 +1,3 @@
---pre
 --extra-index-url https://download.pytorch.org/whl/nightly/cpu
 -r requirements.txt
 torch
diff --git a/requirements.directml.txt b/requirements.directml.txt
index ec53889..97cc5ca 100644
--- a/requirements.directml.txt
+++ b/requirements.directml.txt
@@ -1,4 +1,3 @@
---pre
 -r requirements.txt
 torch-directml
 onnxruntime-directml
\ No newline at end of file
diff --git a/requirements.rocm.txt b/requirements.rocm.txt
index 237449f..de7c9a9 100644
--- a/requirements.rocm.txt
+++ b/requirements.rocm.txt
@@ -1,4 +1,3 @@
---pre
 --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.0
 -r requirements.txt
 torch