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

#2869 - pre-commit changes

Browse Source
This commit is contained in:
Charlie Crane
2024-12-17 12:50:14 +00:00
parent 770896200b
commit dc6f2be209
7 changed files with 33 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/unit_tests/_primaite/_game/_agent/test_probabilistic_agent.py
View File

@@ -65,13 +65,15 @@ def test_probabilistic_agent():
# },
# )
pa_config = {"agent_name":"test_agent",
"action_manager": action_space,
"observation_manager": observation_space,
"reward_function": reward_function,
"agent_settings": {
"action_probabilities": {0: P_DO_NOTHING, 1: P_NODE_APPLICATION_EXECUTE, 2: P_NODE_FILE_DELETE},
}}
pa_config = {
"agent_name": "test_agent",
"action_manager": action_space,
"observation_manager": observation_space,
"reward_function": reward_function,
"agent_settings": {
"action_probabilities": {0: P_DO_NOTHING, 1: P_NODE_APPLICATION_EXECUTE, 2: P_NODE_FILE_DELETE},
},
}
pa = ProbabilisticAgent.from_config(config=pa_config)