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

#3075: Add repeat parameter to config file.

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

3
src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
View File

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

2
src/primaite/simulator/system/applications/red_applications/data_manipulation_bot.py
View File

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