#3075: Add repeat parameter to config file.
This commit is contained in:
@@ -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"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
||||
@@ -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:
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user