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

Pass agent settings from config to agent

Browse Source
This commit is contained in:
Jake Walker
2023-11-20 10:38:01 +00:00
parent 227e73602f
commit 7e0e8a4768
3 changed files with 40 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/primaite/game/session.py
View File

@@ -10,7 +10,13 @@ from pydantic import BaseModel
from primaite import getLogger
from primaite.game.agent.actions import ActionManager
from primaite.game.agent.interface import AbstractAgent, AgentExecutionDefinition, DataManipulationAgent, RandomAgent
from primaite.game.agent.interface import (
AbstractAgent,
AgentExecutionDefinition,
AgentSettings,
DataManipulationAgent,
RandomAgent,
)
from primaite.game.agent.observations import ObservationSpace
from primaite.game.agent.rewards import RewardFunction
from primaite.simulator.network.hardware.base import Link, NIC, Node
@@ -439,6 +445,7 @@ class PrimaiteSession:
rew_function = RewardFunction.from_config(reward_function_cfg, session=sess)
execution_definition = AgentExecutionDefinition(**agent_cfg.get("execution_definition", {}))
agent_settings = AgentSettings(**agent_cfg.get("agent_settings", {}))
# CREATE AGENT
if agent_type == "GreenWebBrowsingAgent":
@@ -449,6 +456,7 @@ class PrimaiteSession:
observation_space=obs_space,
reward_function=rew_function,
execution_definition=execution_definition,
agent_settings=agent_settings,
)
sess.agents.append(new_agent)
elif agent_type == "GATERLAgent":
@@ -458,6 +466,7 @@ class PrimaiteSession:
observation_space=obs_space,
reward_function=rew_function,
execution_definition=execution_definition,
agent_settings=agent_settings,
)
sess.agents.append(new_agent)
sess.rl_agent = new_agent
@@ -468,6 +477,7 @@ class PrimaiteSession:
observation_space=obs_space,
reward_function=rew_function,
execution_definition=execution_definition,
agent_settings=agent_settings,
)
sess.agents.append(new_agent)
else: