mirror of
https://github.com/vale981/ray
synced 2025-03-05 18:11:42 -05:00
[RLlib] Pin Gym Everywhere and turn off gpu for recsim tests (#23452)
This commit is contained in:
parent
ef5b9b87d3
commit
9040f54060
2 changed files with 3 additions and 2 deletions
|
@ -7,6 +7,7 @@ python:
|
|||
pip_packages:
|
||||
- pytest
|
||||
- awscli
|
||||
- gym==0.21.0
|
||||
conda_packages: []
|
||||
|
||||
post_build_cmds:
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Add table
Reference in a new issue