fix optional spacemouse import (#186)

* Update __init__.py * Update __init__.py * Update spacemouse.py
ARISE-Initiative · Feb 10, 2021 · e203530 · e203530
1 parent f123b78
commit e203530
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/robosuite/devices/__init__.py b/robosuite/devices/__init__.py
@@ -1,3 +1,8 @@
 from .device import Device
 from .keyboard import Keyboard
-from .spacemouse import SpaceMouse
+try:
+    from .spacemouse import SpaceMouse
+except ImportError:
+    print("""Unable to load module hid, required to interface with SpaceMouse.\n
+           Only Mac OS X is officially supported. Install the additional\n
+           requirements with `pip install -r requirements-extra.txt`""")
diff --git a/robosuite/devices/spacemouse.py b/robosuite/devices/spacemouse.py
@@ -25,7 +25,7 @@
 except ModuleNotFoundError as exc:
     raise ImportError("Unable to load module hid, required to interface with SpaceMouse. "
                       "Only Mac OS X is officially supported. Install the additional "
-                      "requirements with `pip install -r requirements-ik.txt`") from exc
+                      "requirements with `pip install -r requirements-extra.txt`") from exc
 
 from robosuite.utils.transform_utils import rotation_matrix
 from robosuite.devices import Device