amend

pytorch · Dec 4, 2024 · 53a5239 · 53a5239
1 parent 97722ea
commit 53a5239
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/test/test_libs.py b/test/test_libs.py
@@ -1567,6 +1567,9 @@ def test_habitat(self, envname):
         check_env_specs(env)
 
     def test_from_config(self):
+        import habitat
+
+        cfg = habitat.get_config("benchmark/nav/objectnav/objectnav_hssd-hab.yaml")
         env = HabitatEnv.from_config(cfg)
         check_env_specs(env)
         assert isinstance(env, HabitatEnv)

diff --git a/torchrl/envs/libs/habitat.py b/torchrl/envs/libs/habitat.py
@@ -9,7 +9,7 @@
 from torchrl._utils import _make_ordinal_device
 from torchrl.data.utils import DEVICE_TYPING
 from torchrl.envs.common import EnvBase
-from torchrl.envs.libs.gym import GymEnv, set_gym_backend, GymWrapper
+from torchrl.envs.libs.gym import GymEnv, GymWrapper, set_gym_backend
 from torchrl.envs.utils import _classproperty
 
 _has_habitat = importlib.util.find_spec("habitat") is not None