Move logging utils for vectorized obs mal sim env to ./wrappers

mal-lang · Jan 15, 2025 · fe27e46 · fe27e46
1 parent 536b07f
commit fe27e46
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/malsim/wrappers/malsim_vectorized_obs_env.py b/malsim/wrappers/malsim_vectorized_obs_env.py
@@ -14,17 +14,16 @@
 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,
     MalSimAgent,
     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 → ...rs/malsim_vectorized_obs_logging_utils.py b/malsim/sims/mal_sim_logging_utils.py → ...rs/malsim_vectorized_obs_logging_utils.py