From 0a0cb5de19d043bc25f49dfd38c9bac2e2e19c1c Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Mon, 18 Jul 2022 15:10:05 -0400 Subject: [PATCH] BB: Set maximum depth in depth_1_a_test_configuration --- envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py b/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py index 1c5193fb6..ffd9ed882 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py +++ b/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py @@ -8,6 +8,7 @@ from .utils import ( add_subnets, replace_agent_configuration, replace_propagation_credentials, + set_maximum_depth, ) credentials = ( @@ -53,9 +54,10 @@ def _add_credential_collectors(agent_configuration: AgentConfiguration) -> Agent ) -agent_configuration = _add_exploiters( - _add_subnets(_add_credential_collectors(noop_test_configuration.agent_configuration)) -) +agent_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 1) +agent_configuration = _add_exploiters(agent_configuration) +agent_configuration = _add_subnets(agent_configuration) +agent_configuration = _add_credential_collectors(agent_configuration) depth_1_a_test_configuration = replace_agent_configuration( noop_test_configuration, agent_configuration