Merge branch 'develop' into documentation

inverted-ai · Oct 13, 2022 · 754fe15 · 754fe15
2 parents 80b7928 + 15113f6
commit 754fe15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/invertedai/api_resources.py b/invertedai/api_resources.py
@@ -216,7 +216,7 @@ def initialize(
                 "include_recurrent_states": include_recurrent_states,
             }
             initial_states = iai.session.request(model="initialize", params=params)
-            agents_spawned = len(initial_states["initial_condition"]["agent_states"][0])
+            agents_spawned = len(initial_states["initial_condition"]["agent_states"])
             if agents_spawned != agent_count:
                 iai.logger.warning(
                     f"Unable to spawn a scenario for {agent_count} agents,  {agents_spawned} spawned instead."