From 10652b81c8dbac4a70afb174ba2a3c15a98e1664 Mon Sep 17 00:00:00 2001 From: SangBin Cho Date: Fri, 1 May 2020 00:27:13 -0700 Subject: [PATCH] [BRING BACK TO MASTER] add torch download for rllib regresstion test. --- .../rllib_regresssion_tests/ray-project/project.yaml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ci/regression_test/rllib_regresssion_tests/ray-project/project.yaml b/ci/regression_test/rllib_regresssion_tests/ray-project/project.yaml index 3c5cd2401..7f2db2cbf 100644 --- a/ci/regression_test/rllib_regresssion_tests/ray-project/project.yaml +++ b/ci/regression_test/rllib_regresssion_tests/ray-project/project.yaml @@ -38,7 +38,9 @@ commands: help: "Show tail of the workoad output." - name: run-regression-tests - command: source activate tensorflow_p36 && rllib train -f compact-regression-test.yaml + command: | + source activate tensorflow_p36 && pip install torch torchvision + source activate tensorflow_p36 && rllib train -f compact-regression-test.yaml help: "Run rllib regression tests" # Pathnames for files and directories that should be saved