Update actions and agents to get all tests passing post-refactor

This commit is contained in:
Marek Wolan
2025-01-20 14:07:51 +00:00
parent c521528a0a
commit 18a665e562
5 changed files with 18 additions and 44 deletions

View File

@@ -84,6 +84,7 @@ class ConfigureDoSBotAction(AbstractAction, identifier="configure_dos_bot"):
dos_intensity=config.dos_intensity,
max_sessions=config.max_sessions,
)
data = {k: v for k, v in data.items() if v is not None}
return ["network", "node", config.node_name, "application", "DoSBot", "configure", data]

View File

@@ -16,7 +16,7 @@ __all__ = "ProbabilisticAgent"
class ProbabilisticAgent(AbstractScriptedAgent, identifier="ProbabilisticAgent"):
"""Scripted agent which randomly samples its action space with prescribed probabilities for each action."""
rng: Generator = np.random.default_rng(np.random.randint(0, 65535))
rng: Generator = Field(default_factory=lambda: np.random.default_rng(np.random.randint(0, 65535)))
class AgentSettingsSchema(AbstractScriptedAgent.AgentSettingsSchema):
"""Schema for the `agent_settings` part of the agent config."""