#3075: Add repeat parameter to config file.

This commit is contained in:
Nick Todd
2025-02-03 11:33:21 +00:00
parent 14c9d10aa6
commit 32de95917e
2 changed files with 3 additions and 2 deletions

View File

@@ -405,8 +405,7 @@
"from primaite.session.environment import PrimaiteGymEnv\n",
"from primaite.game.agent.interface import AgentHistoryItem\n",
"import yaml\n",
"from pprint import pprint\n",
"from primaite.game.agent.scripted_agents import probabilistic_agent, data_manipulation_bot\n"
"from pprint import pprint\n"
]
},
{

View File

@@ -52,6 +52,7 @@ class DataManipulationBot(Application, identifier="DataManipulationBot"):
payload: str = "DELETE"
port_scan_p_of_success: float = 0.1
data_manipulation_p_of_success: float = 0.1
repeat: bool = True
config: "DataManipulationBot.ConfigSchema" = Field(default_factory=lambda: DataManipulationBot.ConfigSchema())
@@ -76,6 +77,7 @@ class DataManipulationBot(Application, identifier="DataManipulationBot"):
self.payload = self.config.payload
self.port_scan_p_of_success = self.config.port_scan_p_of_success
self.data_manipulation_p_of_success = self.config.data_manipulation_p_of_success
self.repeat = self.config.repeat
def describe_state(self) -> Dict:
"""