From fe27e46e34395fbca0dcd07ea9eb2a3b8dd409b7 Mon Sep 17 00:00:00 2001 From: Joakim Loxdal Date: Wed, 15 Jan 2025 14:51:32 +0100 Subject: [PATCH] Move logging utils for vectorized obs mal sim env to ./wrappers --- malsim/wrappers/malsim_vectorized_obs_env.py | 5 ++--- .../malsim_vectorized_obs_logging_utils.py} | 0 2 files changed, 2 insertions(+), 3 deletions(-) rename malsim/{sims/mal_sim_logging_utils.py => wrappers/malsim_vectorized_obs_logging_utils.py} (100%) diff --git a/malsim/wrappers/malsim_vectorized_obs_env.py b/malsim/wrappers/malsim_vectorized_obs_env.py index 5416a070..2b2975c9 100644 --- a/malsim/wrappers/malsim_vectorized_obs_env.py +++ b/malsim/wrappers/malsim_vectorized_obs_env.py @@ -14,9 +14,8 @@ import numpy as np from gymnasium.spaces import MultiDiscrete, Box, Dict from pettingzoo import ParallelEnv -from maltoolbox.attackgraph import AttackGraph, AttackGraphNode +from maltoolbox.attackgraph import AttackGraphNode -from ..sims.mal_sim_settings import MalSimulatorSettings from ..sims.mal_simulator import ( MalSimulator, AgentType, @@ -24,7 +23,7 @@ MalSimAttacker, MalSimDefender ) -from ..sims.mal_sim_logging_utils import ( +from .malsim_vectorized_obs_logging_utils import ( format_full_observation, log_mapping_tables, ) diff --git a/malsim/sims/mal_sim_logging_utils.py b/malsim/wrappers/malsim_vectorized_obs_logging_utils.py similarity index 100% rename from malsim/sims/mal_sim_logging_utils.py rename to malsim/wrappers/malsim_vectorized_obs_logging_utils.py