Minor fixes for #47

2024-03-25 18:50:47 +01:00
parent f00252552a
commit c5635b7d94
37 changed files with 13 additions and 18 deletions
--- a/lerobot/common/envs/pusht/env.py
+++ b/lerobot/common/envs/pusht/env.py
@@ -50,7 +50,7 @@ class PushtEnv(AbstractEnv):

    def _make_env(self):
        if not _has_gym:
-            raise ImportError("Cannot import gym.")
+            raise ImportError("Cannot import gymnasium.")

        # TODO(rcadene) (PushTEnv is similar to PushTImageEnv, but without the image rendering, it's faster to iterate on)
        # from lerobot.common.envs.pusht.pusht_env import PushTEnv