From 0651da77540747839c8ea55bd7a15cfc909c4c95 Mon Sep 17 00:00:00 2001 From: Ondrej Lukas Date: Tue, 9 Sep 2025 13:27:07 +0200 Subject: [PATCH 1/2] add seed parameter --- AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py b/AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py index 3b0fdcd2..8b7da1db 100644 --- a/AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py +++ b/AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py @@ -165,6 +165,16 @@ def _create_state_from_view(self, view, add_neighboring_nets = True): type=str, default="netsecenv_conf.yaml", ) + parser.add_argument( + "-s", + "--seed", + help="Random seed for the environment", + action="store", + required=False, + type=int, + default=42, + ) + args = parser.parse_args() print(args) @@ -184,6 +194,6 @@ def _create_state_from_view(self, view, add_neighboring_nets = True): level=pass_level, ) - game_server = WhiteBoxNSGCoordinator(args.game_host, args.game_port, args.task_config) + game_server = WhiteBoxNSGCoordinator(args.game_host, args.game_port, args.task_config, seed=args.seed) # Run it! game_server.run() \ No newline at end of file From 02df43e228e1324504dc6d6cfe80266a9571ee7c Mon Sep 17 00:00:00 2001 From: Ondrej Lukas Date: Tue, 9 Sep 2025 13:27:33 +0200 Subject: [PATCH 2/2] Add neighboring nets by default --- AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py b/AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py index 8b7da1db..91573048 100644 --- a/AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py +++ b/AIDojoCoordinator/worlds/WhiteBoxNSGCoordinator.py @@ -118,7 +118,7 @@ def _generate_all_actions(self)-> list: def _create_state_from_view(self, view, add_neighboring_nets = True): - return super()._create_state_from_view(view, add_neighboring_nets=False) + return super()._create_state_from_view(view, add_neighboring_nets=add_neighboring_nets) if __name__ == "__main__": parser = argparse.ArgumentParser(