Skip to content

Commit eebd2ce

Browse files
authored
Validate poetry for release (#1567)
* Validate poetry for release * test * test * fixtypo
1 parent 07efc44 commit eebd2ce

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

.github/scripts/validate_poetry.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,11 @@ if [[ ${MATRIX_CHANNEL} != "release" ]]; then
2626
fi
2727
else
2828
export PYTHON_KEYRING_BACKEND=keyring.backends.null.Keyring
29+
poetry source add --priority=explicit pytorch "https://download.pytorch.org/whl/${MATRIX_DESIRED_CUDA}"
2930
if [[ ${TORCH_ONLY} == 'true' ]]; then
3031
poetry --quiet add torch
3132
else
32-
poetry --quiet add torch torchaudio torchvision
33+
poetry --quiet add --source pytorch torch torchaudio torchvision
3334
fi
3435
fi
3536

.github/workflows/validate-linux-binaries.yml

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,11 +68,8 @@ jobs:
6868
eval "$(conda shell.bash hook)"
6969
7070
# Special case PyPi installation package. And Install of PyPi package via poetry
71-
if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" ]] && \
72-
([[ ${MATRIX_GPU_ARCH_VERSION} == "12.1" && ${MATRIX_CHANNEL} != "release" ]] || \
73-
[[ ${MATRIX_GPU_ARCH_VERSION} == "11.7" && ${MATRIX_CHANNEL} == "release" ]]); then
71+
if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" && ${MATRIX_GPU_ARCH_VERSION} == "12.1" ]]; then
7472
source ./.github/scripts/validate_pipy.sh
75-
# temporary disable poetry check
7673
source ./.github/scripts/validate_poetry.sh
7774
fi
7875

0 commit comments

Comments
 (0)