Change stretch log levels

huggingface · Sep 26, 2024 · e0fa5e4 · e0fa5e4
1 parent 7c87025
commit e0fa5e4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lerobot/common/robot_devices/robots/stretch.py b/lerobot/common/robot_devices/robots/stretch.py
@@ -20,6 +20,7 @@
 import torch
 from stretch_body.gamepad_teleop import GamePadTeleop
 from stretch_body.robot import Robot as StretchAPI
+from stretch_body.robot_params import RobotParams
 
 from lerobot.common.robot_devices.cameras.utils import Camera
 
@@ -48,8 +49,10 @@ def __init__(self, config: StretchRobotConfig | None = None, **kwargs):
         self.is_connected = False
         self.teleop = None
         self.logs = {}
-        # TODO(aliberts): remove original low-level logging from stretch
-        # RobotParams.set_logging_level("INFO")  # <-- not working
+
+        # TODO(aliberts): test this
+        RobotParams.set_logging_level("WARNING")
+        RobotParams.set_logging_formatter("brief_console_formatter")
 
         self.state_keys = None
         self.action_keys = None