refactor util imports

Total-RD · Nov 23, 2022 · e46b7a4 · e46b7a4
1 parent 8804e66
commit e46b7a4
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/src/pymgrid/envs/discrete/discrete.py b/src/pymgrid/envs/discrete/discrete.py
@@ -5,7 +5,7 @@
 from math import isclose
 
 from pymgrid.envs.base import BaseMicrogridEnv
-from pymgrid.microgrid.utils.logger import ModularLogger
+from pymgrid.utils.logger import ModularLogger
 
 
 class DiscreteMicrogridEnv(BaseMicrogridEnv):

diff --git a/src/pymgrid/modular_microgrid/modular_microgrid.py b/src/pymgrid/modular_microgrid/modular_microgrid.py
@@ -7,9 +7,9 @@
 
 from pymgrid.microgrid import DEFAULT_HORIZON
 from pymgrid.modules import ModuleContainer, UnbalancedEnergyModule
-from pymgrid.microgrid.utils.logger import ModularLogger
-from pymgrid.microgrid.utils.step import MicrogridStep
-from pymgrid.microgrid.utils.serialize import add_numpy_pandas_representers, add_numpy_pandas_constructors, dump_data
+from pymgrid.utils.logger import ModularLogger
+from pymgrid.modular_microgrid.utils.step import MicrogridStep
+from pymgrid.utils.serialize import add_numpy_pandas_representers, add_numpy_pandas_constructors, dump_data
 
 
 class Microgrid(yaml.YAMLObject):

diff --git a/src/pymgrid/modules/base/base_module.py b/src/pymgrid/modules/base/base_module.py
@@ -7,9 +7,9 @@
 from gym.spaces import Box
 from warnings import warn
 
-from pymgrid.microgrid.utils.logger import ModularLogger
-from pymgrid.microgrid.utils.normalize import Normalize, IdentityNormalize
-from pymgrid.microgrid.utils.serialize import add_numpy_pandas_representers, add_numpy_pandas_constructors, dump_data
+from pymgrid.utils.logger import ModularLogger
+from pymgrid.modules.utils.normalize import Normalize, IdentityNormalize
+from pymgrid.utils.serialize import add_numpy_pandas_representers, add_numpy_pandas_constructors, dump_data
 
 
 script_logger = logging.getLogger(__name__)