From eb91619175a133c8bafd57882fffd8a4c8e6bff1 Mon Sep 17 00:00:00 2001 From: Sven Mika Date: Wed, 29 Apr 2020 10:36:41 +0200 Subject: [PATCH] Fix release 0.8.5 tests for PPO torch Breakout. (#8226) --- .../rllib_regresssion_tests/compact-regression-test.yaml | 2 +- rllib/agents/ppo/ppo.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ci/regression_test/rllib_regresssion_tests/compact-regression-test.yaml b/ci/regression_test/rllib_regresssion_tests/compact-regression-test.yaml index 4dbb2a15b..83e48d204 100644 --- a/ci/regression_test/rllib_regresssion_tests/compact-regression-test.yaml +++ b/ci/regression_test/rllib_regresssion_tests/compact-regression-test.yaml @@ -52,7 +52,7 @@ atari-ppo-torch: stop: time_total_s: 3600 config: - use_pytorch: true, + use_pytorch: true lambda: 0.95 kl_coeff: 0.5 clip_rewards: True diff --git a/rllib/agents/ppo/ppo.py b/rllib/agents/ppo/ppo.py index 4df6a3362..0a60ee416 100644 --- a/rllib/agents/ppo/ppo.py +++ b/rllib/agents/ppo/ppo.py @@ -179,7 +179,7 @@ def validate_config(config): def get_policy_class(config): - if config.get("use_pytorch") is True: + if config["use_pytorch"]: from ray.rllib.agents.ppo.ppo_torch_policy import PPOTorchPolicy return PPOTorchPolicy else: