tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

chore(style): fix format

Browse Source
This commit is contained in:
Steven Palma
2025-03-07 15:14:46 +01:00
parent 0eb56cec14
commit f9e4a1f5c4
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lerobot/configs/policies.py
View File

@@ -64,13 +64,14 @@ class PreTrainedConfig(draccus.ChoiceRegistry, HubMixin, abc.ABC):
self.pretrained_path = None self.pretrained_path = None
if not self.device or not is_torch_device_available(self.device): if not self.device or not is_torch_device_available(self.device):
auto_device = auto_select_torch_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 self.device = auto_device.type
# Automatically deactivate AMP if necessary # Automatically deactivate AMP if necessary
if self.use_amp and not is_amp_available(self.device): if self.use_amp and not is_amp_available(self.device):
logging.warning( 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 self.use_amp = False

1
lerobot/scripts/control_sim_robot.py
View File

@@ -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. **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. # TODO(Steven): This script should be updated to use the new robot API and the new dataset API.
import argparse import argparse
import importlib import importlib