From 7b6bf8f1620a72204298bd3361cd29d0223ed65f Mon Sep 17 00:00:00 2001 From: Ervin Teng Date: Wed, 14 Aug 2019 12:55:58 -0700 Subject: [PATCH] Only call get_action on brains with policies --- ml-agents-envs/mlagents/envs/subprocess_env_manager.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ml-agents-envs/mlagents/envs/subprocess_env_manager.py b/ml-agents-envs/mlagents/envs/subprocess_env_manager.py index f6dadc22a6..13c3851c91 100644 --- a/ml-agents-envs/mlagents/envs/subprocess_env_manager.py +++ b/ml-agents-envs/mlagents/envs/subprocess_env_manager.py @@ -251,7 +251,8 @@ def _postprocess_steps( def _take_step(self, last_step: StepInfo) -> Dict[str, ActionInfo]: all_action_info: Dict[str, ActionInfo] = {} for brain_name, brain_info in last_step.current_all_brain_info.items(): - all_action_info[brain_name] = self.policies[brain_name].get_action( - brain_info - ) + if brain_name in self.policies: + all_action_info[brain_name] = self.policies[brain_name].get_action( + brain_info + ) return all_action_info