frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move all ONNX test dependencies to Docker (#96590)

Browse Source 
Per title.  This is the first one of a two-part process:

[x] Move all ONNX test dependencies to Docker https://github.com/pytorch/pytorch/pull/96590
[ ] Move the test model used by [TestFxToOnnxWithOnnxRuntime.test_gpt2_tiny](https://hud.pytorch.org/failure/FAILED%20test%2Fonnx%2Ftest_fx_to_onnx_with_onnxruntime.py%3A%3ATestFxToOnnxWithOnnxRuntime%3A%3Atest_large_scale_exporter_with_tiny_gpt2%20-%20requests.exceptions.ReadTimeout%3A%20HTTPSConnectionPool(host%3D'huggingface.co'%2C%20port%3D443)%3A%20Read%20timed%20out.%20(read%20timeout%3D10.0))
Pull Request resolved: https://github.com/pytorch/pytorch/pull/96590
Approved by: https://github.com/ZainRizvi
This commit is contained in:
Huy Do
2023-03-14 06:19:00 +00:00
committed by PyTorch MergeBot
parent 70090b4daf
commit 6a2dcfd738
6 changed files with 37 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/onnx/test.sh
View File

@ -24,15 +24,6 @@ do
done
set -- "${UNKNOWN[@]}" # leave UNKNOWN
if [[ $PARALLEL == 1 ]]; then
pip install pytest-xdist
fi
# pytest, scipy, hypothesis: these may not be necessary
# pytest-cov: installing since `coverage run -m pytest ..` doesn't work
# pytest-subtests: unittest subtests support for pytest
# parameterized: parameterizing test class
pip install pytest scipy hypothesis pytest-cov pytest-subtests parameterized
# allows coverage to run w/o failing due to a missing plug-in
pip install -e tools/coverage_plugins_package