Fixes for CUDA CI (#3413)

2026-04-14 23:52:52 -07:00
parent 4400504ad5
commit 57bcced8cb
4 changed files with 6 additions and 2 deletions
@@ -18,7 +18,7 @@ runs:
      env:
        CMAKE_ARGS: -DMLX_BUILD_CUDA=ON
      run: |
-        pip install auditwheel build patchelf setuptools
+        pip install auditwheel "build<=1.4.2" patchelf setuptools
        python setup.py clean --all
        MLX_DISABLE_SM90A_KERNELS=1 MLX_BUILD_STAGE=2 python -m build -w

@@ -25,7 +25,7 @@ runs:
    - name: Build Python package
      shell: bash
      run: |
-        pip install auditwheel patchelf build
+        pip install auditwheel patchelf "build<=1.4.2"
        python setup.py clean --all
        MLX_BUILD_STAGE=1 python -m build -w
        auditwheel repair dist/mlx-*.whl \