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: