make key as string (facebookresearch#1764)

Co-authored-by: Jimmy Yang <[email protected]>
SgtVincent · Jan 17, 2024 · b813819 · b813819
1 parent 39c19b7
commit b813819
Showing 1 changed file with 1 addition and 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
@@ -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: