refactor: frameworks/environments/habitat.py -> simulators/habitat/en…

…vironment.py
thousandbrainsproject · Jan 23, 2025 · 24ae451 · 24ae451
1 parent e5d528e
commit 24ae451
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/src/tbp/monty/frameworks/environments/two_d_data.py b/src/tbp/monty/frameworks/environments/two_d_data.py
@@ -21,7 +21,7 @@
 from tbp.monty.frameworks.actions.actions import Action
 from tbp.monty.frameworks.environment_utils.transforms import DepthTo3DLocations
 from tbp.monty.frameworks.environments.embodied_environment import EmbodiedEnvironment
-from tbp.monty.frameworks.environments.habitat import HabitatActionSpace
+from tbp.monty.simulators.habitat.environment import HabitatActionSpace
 
 __all__ = [
     "OmniglotEnvironment",

diff --git a/src/tbp/monty/simulators/habitat/configs.py b/src/tbp/monty/simulators/habitat/configs.py
@@ -27,12 +27,12 @@
     DepthTo3DLocations,
     MissingToMaxDepth,
 )
-from tbp.monty.frameworks.environments.habitat import (
+from tbp.monty.simulators.habitat import MultiSensorAgent, SingleSensorAgent
+from tbp.monty.simulators.habitat.environment import (
     AgentConfig,
     HabitatEnvironment,
     ObjectConfig,
 )
-from tbp.monty.simulators.habitat import MultiSensorAgent, SingleSensorAgent
 
 
 @dataclass

diff --git a/.../monty/frameworks/environments/habitat.py → ...p/monty/simulators/habitat/environment.py b/.../monty/frameworks/environments/habitat.py → ...p/monty/simulators/habitat/environment.py
diff --git a/tests/unit/habitat_data_test.py b/tests/unit/habitat_data_test.py
@@ -23,9 +23,9 @@
     EnvironmentDataset,
 )
 from tbp.monty.frameworks.environments.embodied_environment import ActionSpace
-from tbp.monty.frameworks.environments.habitat import AgentConfig, HabitatEnvironment
 from tbp.monty.frameworks.models.motor_policies import BasePolicy
 from tbp.monty.simulators.habitat import SingleSensorAgent
+from tbp.monty.simulators.habitat.environment import AgentConfig, HabitatEnvironment
 
 DATASET_LEN = 10
 DEFAULT_ACTUATION_AMOUNT = 0.25