p41275903
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

BB: Add set_maximum_depth()

This commit is contained in:
Mike Salvatore 2022-07-18 15:03:52 -04:00
parent c2028f15a4
commit 5a1a40a515
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
envs/monkey_zoo/blackbox/test_configurations/utils.py
View File

@ -55,6 +55,15 @@ def add_credential_collectors(
return replace(agent_configuration, credential_collectors=tuple(credential_collectors))
def set_maximum_depth(
agent_configuration: AgentConfiguration, maximum_depth: int
) -> AgentConfiguration:
propagation_configuration = replace(
agent_configuration.propagation, maximum_depth=maximum_depth
)
return replace_propagation_configuration(agent_configuration, propagation_configuration)
def replace_exploitation_configuration(
agent_configuration: AgentConfiguration, exploitation_configuration: ExploitationConfiguration
) -> AgentConfiguration: