Make keyboard optional

hanruihua · Dec 22, 2024 · c574c1c · c574c1c
1 parent 5ea4dc0
commit c574c1c
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/irsim/env/env_base.py b/irsim/env/env_base.py
@@ -24,8 +24,9 @@
 
 try:
     from pynput import keyboard
+    keyboard_module = True
 except ImportError:
-    print("pynput is not installed, keyboard control is disabled.")
+    keyboard_module = False
 
 class EnvBase:
     """
@@ -87,7 +88,13 @@ def __init__(
         env_param.objects = self.objects
 
         if world_param.control_mode == "keyboard":
-            self.init_keyboard(self.env_config.parse["keyboard"])
+            if not keyboard_module:
+                self.logger.error(
+                    "Keyboard module is not installed. Auto control applied. Please install pynput by 'pip install pynput'."
+                )
+                world_param.control_mode = "auto"
+            else:
+                self.init_keyboard(self.env_config.parse["keyboard"])
 
         if full:
             system_platform = platform.system()