From 6da9d859aa6c735eabc1f86ebec581ffc86b3f6d Mon Sep 17 00:00:00 2001 From: Kourosh Hakhamaneshi Date: Sun, 26 Mar 2023 18:42:02 -0700 Subject: [PATCH] [RLlib][RLModule] Disabled RLModule API on cartpole_ppo_fake_gpu tests becauase of LSTMs Signed-off-by: Kourosh Hakhamaneshi --- rllib/tuned_examples/ppo/cartpole-ppo-fake-gpus.yaml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/rllib/tuned_examples/ppo/cartpole-ppo-fake-gpus.yaml b/rllib/tuned_examples/ppo/cartpole-ppo-fake-gpus.yaml index adc638c6d7a1..08b47c188ff8 100644 --- a/rllib/tuned_examples/ppo/cartpole-ppo-fake-gpus.yaml +++ b/rllib/tuned_examples/ppo/cartpole-ppo-fake-gpus.yaml @@ -27,3 +27,6 @@ cartpole-ppo-fake-gpus: # Fake 2 GPUs. num_gpus: 2 _fake_gpus: true + # TODO (Kourosh): Enable RLModule once LSTM is supported + _enable_rl_module_api: false + _enable_learner_api: false