Fix failing Travis build for tvt/run.sh

* Restrict the script to a single episode to avoid timeouts
* Ensure that `wheel` is up to date (we may need to build our own wheels for some of the requirements)

PiperOrigin-RevId: 367980522
This commit is contained in:
Alistair Muldal
2021-04-12 13:05:11 +01:00
committed by Diego de Las Casas
parent 83c201f8e5
commit 0b2370819b
2 changed files with 10 additions and 8 deletions
+6 -5
View File
@@ -8,18 +8,19 @@ python:
- "3.6" - "3.6"
env: env:
matrix: matrix:
- PROJECT="tvt" - PROJECT="avae"
- PROJECT="cs_gan" - PROJECT="cs_gan"
- PROJECT="iodine" - PROJECT="iodine"
- PROJECT="transporter"
- PROJECT="polygen"
- PROJECT="learning_to_simulate"
- PROJECT="avae"
- PROJECT="kfac_ferminet_alpha" - PROJECT="kfac_ferminet_alpha"
- PROJECT="learning_to_simulate"
- PROJECT="polygen"
- PROJECT="transporter"
- PROJECT="tvt"
before_script: before_script:
- sudo apt-get update -qq - sudo apt-get update -qq
- pip install --upgrade pip - pip install --upgrade pip
- pip install --upgrade virtualenv - pip install --upgrade virtualenv
- pip install --upgrade wheel
script: script:
- ./$PROJECT/run.sh - ./$PROJECT/run.sh
+4 -3
View File
@@ -13,8 +13,9 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
python3 -m venv tvt_venv python3 -m venv /tmp/tvt_venv
source tvt_venv/bin/activate source /tmp/tvt_venv/bin/activate
pip install -U pip
pip install -r tvt/requirements.txt pip install -r tvt/requirements.txt
python3 -m tvt.main python3 -m tvt.main --num_episodes=1