diff --git a/.github/scripts/validate_poetry.sh b/.github/scripts/validate_poetry.sh index 741c4d13c..6b7fe2412 100644 --- a/.github/scripts/validate_poetry.sh +++ b/.github/scripts/validate_poetry.sh @@ -22,7 +22,7 @@ fi if [[ ${TORCH_ONLY} == 'true' ]]; then poetry --quiet add torch${RELEASE_SUFFIX} else - poetry --quiet add --source pytorch torch${RELEASE_SUFFIX} torchaudio torchvision + poetry --quiet add torch${RELEASE_SUFFIX} torchaudio torchvision fi python ../test/smoke_test/smoke_test.py ${TEST_SUFFIX} --runtime-error-check disabled