File tree Expand file tree Collapse file tree 2 files changed +4
-17
lines changed Expand file tree Collapse file tree 2 files changed +4
-17
lines changed Original file line number Diff line number Diff line change 1
- conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} numpy
2
- conda activate ${ENV_NAME}
3
- export CONDA_CHANNEL=" pytorch"
4
- export PIP_DOWNLOAD_URL=" https://download.pytorch.org/whl/cpu"
5
- export TEXT_PIP_PREFIX=" "
6
-
7
- if [[ ${MATRIX_CHANNEL} = " nightly" ]]; then
8
- export TEXT_PIP_PREFIX=" --pre"
9
- export PIP_DOWNLOAD_URL=" https://download.pytorch.org/whl/nightly/cpu"
10
- export CONDA_CHANNEL=" pytorch-nightly"
11
- elif [[ ${MATRIX_CHANNEL} = " test" ]]; then
12
- export PIP_DOWNLOAD_URL=" https://download.pytorch.org/whl/test/cpu"
13
- export CONDA_CHANNEL=" pytorch-test"
14
- fi
15
1
16
2
if [[ ${MATRIX_PACKAGE_TYPE} = " conda" ]]; then
17
- conda install -y torchtext pytorch -c ${CONDA_CHANNEL }
3
+ conda install -y torchtext -c ${PYTORCH_CONDA_CHANNEL }
18
4
else
19
- pip install ${TEXT_PIP_PREFIX } torchtext torch --extra-index-url ${PIP_DOWNLOAD_URL }
5
+ pip install ${PYTORCH_PIP_PREFIX } torchtext --extra-index-url ${PYTORCH_PIP_DOWNLOAD_URL }
20
6
fi
21
7
22
8
python ./test/smoke_tests/smoke_tests.py
Original file line number Diff line number Diff line change 45
45
type : string
46
46
jobs :
47
47
validate-binaries :
48
- uses : pytorch/builder /.github/workflows/validate-domain-library.yml@main
48
+ uses : pytorch/test-infra /.github/workflows/validate-domain-library.yml@main
49
49
with :
50
50
package_type : " conda,wheel"
51
51
os : ${{ inputs.os }}
52
52
channel : ${{ inputs.channel }}
53
53
repository : " pytorch/text"
54
54
smoke_test : " source ./.github/scripts/validate_binaries.sh"
55
+ install_torch : true
You can’t perform that action at this time.
0 commit comments