File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -19,16 +19,14 @@ if [[ ! -z ${RELEASE_VERSION} ]]; then
19
19
RELEASE_SUFFIX=" @${RELEASE_VERSION} "
20
20
fi
21
21
22
- # Installing poetry from our custom repo. We need to configure it before use and disable authentication
23
- export PYTHON_KEYRING_BACKEND=keyring.backends.null.Keyring
24
- poetry source add --priority=explicit pytorch " https://download.pytorch.org/whl/${MATRIX_DESIRED_CUDA} "
22
+
25
23
if [[ ${TORCH_ONLY} == ' true' ]]; then
26
24
poetry --quiet add torch${RELEASE_SUFFIX}
27
25
else
28
26
poetry --quiet add --source pytorch torch${RELEASE_SUFFIX} torchaudio torchvision
29
27
fi
30
28
31
- python ../test/smoke_test/smoke_test.py ${TEST_SUFFIX} --runtime-error-check disabled
29
+ python ../test/smoke_test/smoke_test.py --package torchonly --runtime-error-check disabled
32
30
conda deactivate
33
31
conda env remove -p ${ENV_NAME} _poetry
34
32
cd ..
You can’t perform that action at this time.
0 commit comments