Import error update

Author: kevin-thankyou-lin

robosuite/devices/keyboard.py

@@ -3,7 +3,10 @@
 """
 
 import numpy as np
-from pynput.keyboard import Controller, Key, Listener
+try:
+    from pynput.keyboard import Controller, Key, Listener
+except ImportError as e:
+    print(f"Error: {e}.")
 
 from robosuite.devices import Device
 from robosuite.utils.transform_utils import rotation_matrix

robosuite/devices/spacemouse.py

@@ -21,20 +21,21 @@
 from collections import namedtuple
 
 import numpy as np
-from pynput.keyboard import Controller, Key, Listener
+try:
+    from pynput.keyboard import Controller, Key, Listener
+except ImportError as exc:
+    print(f"Unable to load module pynput, required to interface with SpaceMouse. Error: {exc}")
 
 from robosuite.utils.log_utils import ROBOSUITE_DEFAULT_LOGGER
 
 try:
     import hid
 except ModuleNotFoundError as exc:
-    raise ImportError(
+    print(
         "Unable to load module hid, required to interface with SpaceMouse. "
         "Only macOS is officially supported. Install the additional "
         "requirements with `pip install -r requirements-extra.txt`"
-    ) from exc
-
-from pynput.keyboard import Controller, Key, Listener
+    )
 
 import robosuite.macros as macros
 from robosuite.devices import Device
@@ -371,9 +372,9 @@ def _postprocess_device_outputs(self, dpos, drotation):
         return dpos, drotation
 
 
-if __name__ == "__main__":
+# if __name__ == "__main__":
 
-    space_mouse = SpaceMouse()
-    for i in range(100):
-        print(space_mouse.control, space_mouse.control_gripper)
-        time.sleep(0.02)
+    # space_mouse = SpaceMouse()
+    # for i in range(100):
+    #     print(space_mouse.control, space_mouse.control_gripper)
+    #     time.sleep(0.02)