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

fix(lerobot/common/utils): remove lint warnings/errors

Browse Source
This commit is contained in:
Steven Palma
2025-03-07 14:54:43 +01:00
parent dffa5a18db
commit 85214ec303
5 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/utils/benchmark.py
View File

@@ -68,9 +68,9 @@ class TimeBenchmark(ContextDecorator):
Block took approximately 10.00 milliseconds Block took approximately 10.00 milliseconds
""" """
def __init__(self, print=False): def __init__(self, print_time=False):
self.local = threading.local() self.local = threading.local()
self.print_time = print self.print_time = print_time
def __enter__(self): def __enter__(self):
self.local.start_time = time.perf_counter() self.local.start_time = time.perf_counter()

2
lerobot/common/utils/import_utils.py
View File

@@ -46,7 +46,7 @@ def is_package_available(pkg_name: str, return_version: bool = False) -> tuple[b
else: else:
# For packages other than "torch", don't attempt the fallback and set as not available # For packages other than "torch", don't attempt the fallback and set as not available
package_exists = False package_exists = False
logging.debug(f"Detected {pkg_name} version: {package_version}") logging.debug("Detected %s version: %s", {pkg_name}, package_version)
if return_version: if return_version:
return package_exists, package_version return package_exists, package_version
else: else:

2
lerobot/common/utils/logging_utils.py
View File

@@ -27,6 +27,8 @@ class AverageMeter:
def __init__(self, name: str, fmt: str = ":f"): def __init__(self, name: str, fmt: str = ":f"):
self.name = name self.name = name
self.fmt = fmt self.fmt = fmt
self.val = 0.0
self.avg = 0.0
self.reset() self.reset()
def reset(self) -> None: def reset(self) -> None:

2
lerobot/common/utils/utils.py
View File

@@ -69,7 +69,7 @@ def get_safe_torch_device(try_device: str, log: bool = False) -> torch.device:
case _: case _:
device = torch.device(try_device) device = torch.device(try_device)
if log: if log:
logging.warning(f"Using custom {try_device} device.") logging.warning("Using custom %s device.", try_device)
return device return device

4
lerobot/common/utils/wandb_utils.py
View File

@@ -86,7 +86,7 @@ class WandBLogger:
resume="must" if cfg.resume else None, resume="must" if cfg.resume else None,
) )
print(colored("Logs will be synced with wandb.", "blue", attrs=["bold"])) print(colored("Logs will be synced with wandb.", "blue", attrs=["bold"]))
logging.info(f"Track this run --> {colored(wandb.run.get_url(), 'yellow', attrs=['bold'])}") logging.info("Track this run --> %s", colored(wandb.run.get_url(), "yellow", attrs=["bold"]))
self._wandb = wandb self._wandb = wandb
def log_policy(self, checkpoint_dir: Path): def log_policy(self, checkpoint_dir: Path):
@@ -108,7 +108,7 @@ class WandBLogger:
for k, v in d.items(): for k, v in d.items():
if not isinstance(v, (int, float, str)): if not isinstance(v, (int, float, str)):
logging.warning( logging.warning(
f'WandB logging of key "{k}" was ignored as its type is not handled by this wrapper.' 'WandB logging of key "%s" was ignored as its type is not handled by this wrapper.', k
) )
continue continue
self._wandb.log({f"{mode}/{k}": v}, step=step) self._wandb.log({f"{mode}/{k}": v}, step=step)