From 2177e2f410439c10b38e728aec5fba647c226489 Mon Sep 17 00:00:00 2001 From: opherlieber Date: Thu, 3 Jan 2019 08:19:59 +0200 Subject: [PATCH] [rllib] Agent: Allow unknown subkeys for custom_resources_per_worker (#3639) * RLLib Agent: Allow unknown subkeys for custom_resources_per_worker * Update agent.py --- python/ray/rllib/agents/agent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/ray/rllib/agents/agent.py b/python/ray/rllib/agents/agent.py index 6f151b8e0..26981a4c9 100644 --- a/python/ray/rllib/agents/agent.py +++ b/python/ray/rllib/agents/agent.py @@ -201,7 +201,8 @@ class Agent(Trainable): _allow_unknown_configs = False _allow_unknown_subkeys = [ - "tf_session_args", "env_config", "model", "optimizer", "multiagent" + "tf_session_args", "env_config", "model", "optimizer", "multiagent", + "custom_resources_per_worker" ] def __init__(self, config=None, env=None, logger_creator=None):