Add call validation

2024-03-08 15:57:43 +00:00
parent beb51834f9
commit 0447a05084
9 changed files with 69 additions and 44 deletions
--- a/src/primaite/session/io.py
+++ b/src/primaite/session/io.py
@@ -93,7 +93,7 @@ class PrimaiteIO:
                {
                    "episode": episode,
                    "timestep": timestep,
-                    "agent_actions": {k: {"action": v[0], "parameters": v[1]} for k, v in agent_actions.items()},
+                    "agent_actions": agent_actions,
                }
            ]
        )