Pin setuptools for old PyTorch

MiroPsota · Jun 28, 2024 · 745b431 · 745b431
1 parent b39bc34
commit 745b431
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/.github/workflows/build_wheels.yml b/.github/workflows/build_wheels.yml
@@ -136,9 +136,19 @@ jobs:
 
       - name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.compute-platform }}
         run: |
+          versionlte() {
+            [ "$1" = "`echo -e "$1\n$2" | sort -V | head -n1`" ]
+          }
+          versionlt() {
+            [ "$1" = "$2" ] && return 1 || versionlte $1 $2
+          }
+          VERSION_LIMIT=""
+          if versionlt ${{ matrix.torch-version }} "2.2.0" ; then
+            VERSION_LIMIT="<70"
+          fi
           pip install --upgrade \
             wheel \
-            setuptools \
+            setuptools$VERSION_LIMIT \
             build \
             --extra-index-url https://download.pytorch.org/whl/${{ matrix.compute-platform }} \
             torch==${{ matrix.torch-version }} \