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

#2869 - Update type hints and ConfigSchema variables in some agent classes

Browse Source
This commit is contained in:
Charlie Crane
2025-01-13 15:08:48 +00:00
parent 32fc970cfe
commit edd2668ea4
4 changed files with 33 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,7 +3,7 @@ from primaite.game.agent.actions import ActionManager
from primaite.game.agent.observations.observation_manager import NestedObservation, ObservationManager
from primaite.game.agent.rewards import RewardFunction
from primaite.game.agent.scripted_agents.probabilistic_agent import ProbabilisticAgent
from primaite.game.game import PrimaiteGame
from primaite.game.game import PrimaiteGame, PrimaiteGameOptions
def test_probabilistic_agent():
@@ -54,16 +54,17 @@ def test_probabilistic_agent():
},
"options": {},
}
observation_space = ObservationManager(NestedObservation(components={}))
reward_function = RewardFunction()
game = PrimaiteGame()
game.options = PrimaiteGameOptions(ports=[], protocols=[])
observation_space_cfg = None
reward_function_cfg = {}
pa_config = {
"agent_name": "test_agent",
"game": PrimaiteGame(),
"type": "ProbabilisticAgent",
"game": game,
"action_manager": action_space_cfg,
"observation_manager": observation_space_cfg,
"reward_function": reward_function_cfg,