From 9040f54060376bd594562cc752053b730ba4954e Mon Sep 17 00:00:00 2001
From: Avnish Narayan <38871737+avnishn@users.noreply.github.com>
Date: Thu, 24 Mar 2022 01:17:30 -0700
Subject: [PATCH] [RLlib] Pin Gym Everywhere and turn off gpu for recsim tests
 (#23452)

---
 release/long_running_distributed_tests/app_config.yaml      | 1 +
 release/rllib_tests/learning_tests/hard_learning_tests.yaml | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/release/long_running_distributed_tests/app_config.yaml b/release/long_running_distributed_tests/app_config.yaml
index 3eb088030..59e1fd010 100644
--- a/release/long_running_distributed_tests/app_config.yaml
+++ b/release/long_running_distributed_tests/app_config.yaml
@@ -7,6 +7,7 @@ python:
   pip_packages:
     - pytest
     - awscli
+    - gym==0.21.0
   conda_packages: []
 
 post_build_cmds:
diff --git a/release/rllib_tests/learning_tests/hard_learning_tests.yaml b/release/rllib_tests/learning_tests/hard_learning_tests.yaml
index fac6d307d..2277ab364 100644
--- a/release/rllib_tests/learning_tests/hard_learning_tests.yaml
+++ b/release/rllib_tests/learning_tests/hard_learning_tests.yaml
@@ -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