From f9e4a1f5c4c9f6c5f5d72b1b2b003e161dd8fe88 Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Fri, 7 Mar 2025 15:14:46 +0100 Subject: [PATCH] chore(style): fix format --- lerobot/configs/policies.py | 5 +++-- lerobot/scripts/control_sim_robot.py | 1 + 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/lerobot/configs/policies.py b/lerobot/configs/policies.py index a7628aa8..1a6b34b2 100644 --- a/lerobot/configs/policies.py +++ b/lerobot/configs/policies.py @@ -64,13 +64,14 @@ class PreTrainedConfig(draccus.ChoiceRegistry, HubMixin, abc.ABC): self.pretrained_path = None if not self.device or not is_torch_device_available(self.device): auto_device = auto_select_torch_device() - logging.warning("Device '%s' is not available. Switching to '%s'.",self.device,auto_device) + logging.warning("Device '%s' is not available. Switching to '%s'.", self.device, auto_device) self.device = auto_device.type # Automatically deactivate AMP if necessary if self.use_amp and not is_amp_available(self.device): logging.warning( - "Automatic Mixed Precision (amp) is not available on device '%s'. Deactivating AMP.",self.device + "Automatic Mixed Precision (amp) is not available on device '%s'. Deactivating AMP.", + self.device, ) self.use_amp = False diff --git a/lerobot/scripts/control_sim_robot.py b/lerobot/scripts/control_sim_robot.py index 538e8cdb..a727b00f 100644 --- a/lerobot/scripts/control_sim_robot.py +++ b/lerobot/scripts/control_sim_robot.py @@ -80,6 +80,7 @@ This might require a sudo permission to allow your terminal to monitor keyboard **NOTE**: You can resume/continue data recording by running the same data recording command twice. """ + # TODO(Steven): This script should be updated to use the new robot API and the new dataset API. import argparse import importlib