Validate poetry for release (#1567)

* Validate poetry for release * test * test * fixtypo
pytorch · Oct 6, 2023 · eebd2ce · eebd2ce
1 parent 07efc44
commit eebd2ce
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/.github/scripts/validate_poetry.sh b/.github/scripts/validate_poetry.sh
@@ -26,10 +26,11 @@ if [[ ${MATRIX_CHANNEL} != "release" ]]; then
     fi
 else
     export PYTHON_KEYRING_BACKEND=keyring.backends.null.Keyring
+    poetry source add --priority=explicit pytorch "https://download.pytorch.org/whl/${MATRIX_DESIRED_CUDA}"
     if [[ ${TORCH_ONLY} == 'true' ]]; then
         poetry --quiet add torch
     else
-        poetry --quiet add torch torchaudio torchvision
+        poetry --quiet add --source pytorch torch torchaudio torchvision
     fi
 fi
 

diff --git a/.github/workflows/validate-linux-binaries.yml b/.github/workflows/validate-linux-binaries.yml
@@ -68,11 +68,8 @@ jobs:
         eval "$(conda shell.bash hook)"
 
         # Special case PyPi installation package. And Install of PyPi package via poetry
-        if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" ]] && \
-           ([[ ${MATRIX_GPU_ARCH_VERSION} == "12.1" && ${MATRIX_CHANNEL} != "release" ]] || \
-            [[ ${MATRIX_GPU_ARCH_VERSION} == "11.7" && ${MATRIX_CHANNEL} == "release" ]]); then
+        if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" && ${MATRIX_GPU_ARCH_VERSION} == "12.1" ]]; then
           source ./.github/scripts/validate_pipy.sh
-          # temporary disable poetry check
           source ./.github/scripts/validate_poetry.sh
         fi