From b813819fe0006fc88ac07b32d13531fda719d94b Mon Sep 17 00:00:00 2001 From: Jimmy Yang <55121504+jimmytyyang@users.noreply.github.com> Date: Wed, 17 Jan 2024 11:31:11 -0500 Subject: [PATCH] make key as string (#1764) Co-authored-by: Jimmy Yang --- .../habitat_baselines/rl/multi_agent/multi_agent_access_mgr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/habitat-baselines/habitat_baselines/rl/multi_agent/multi_agent_access_mgr.py b/habitat-baselines/habitat_baselines/rl/multi_agent/multi_agent_access_mgr.py index 511a68d21b..facb74c0f2 100644 --- a/habitat-baselines/habitat_baselines/rl/multi_agent/multi_agent_access_mgr.py +++ b/habitat-baselines/habitat_baselines/rl/multi_agent/multi_agent_access_mgr.py @@ -331,7 +331,7 @@ def load_state_dict(self, state): for agent_i, agent in enumerate(self._agents): if not agent.actor_critic.should_load_agent_state: continue - agent.load_state_dict(state[agent_i]) + agent.load_state_dict(state[str(agent_i)]) def load_ckpt_state_dict(self, ckpt): for agent in self._agents: