[RLlib] Pin Gym Everywhere and turn off gpu for recsim tests (#23452)

This commit is contained in:
Avnish Narayan 2022-03-24 01:17:30 -07:00 committed by GitHub
parent ef5b9b87d3
commit 9040f54060
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions

View file

@ -7,6 +7,7 @@ python:
pip_packages:
- pytest
- awscli
- gym==0.21.0
conda_packages: []
post_build_cmds:

View file

@ -472,7 +472,7 @@ slateq-interest-evolution-recsim-env:
# test for now, we'll use pytorch, and investigate why tensorflow is not working.
# One indication of the issue is that the loss inputs are not flattened from the
# original dictionary spaces.
framework: torch
framework: tf
# Each step, sample `num_candidates` documents using the env-internal
# document sampler model (a logic that creates n documents to select
# the slate from).
@ -487,7 +487,7 @@ slateq-interest-evolution-recsim-env:
convert_to_discrete_action_space: false
seed: 0
num_gpus: 1
num_gpus: 0
exploration_config:
warmup_timesteps: 20000