oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2869 - Edit test fixture to work with new agent system

Browse Source
This commit is contained in:
Marek Wolan
2025-01-14 10:58:34 +00:00
parent e7cfeeafc0
commit 9f5e16dd85
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/conftest.py
View File

@@ -500,10 +500,12 @@ def game_and_agent():
reward_function = RewardFunction() reward_function = RewardFunction()
config = { config = {
"type": "ControlledAgent",
"agent_name": "test_agent", "agent_name": "test_agent",
"action_manager": action_space, "action_manager": action_space,
"observation_manager": observation_space, "observation_manager": observation_space,
"reward_function": reward_function, "reward_function": reward_function,
"agent_settings": {},
} }
test_agent = ControlledAgent.from_config(config=config) test_agent = ControlledAgent.from_config(config=config)