admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

style(wandb): move training example imports to module scope

Browse Source
This commit is contained in:
Pepijn
2026-07-15 13:41:39 +02:00
parent 94dc85b443
commit 87585195e6
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/common/wandb_utils.py
+2 -5
View File
@@ -19,6 +19,8 @@ import re
from glob import glob from glob import glob
from pathlib import Path from pathlib import Path
import numpy as np
import torch
from huggingface_hub.constants import SAFETENSORS_SINGLE_FILE from huggingface_hub.constants import SAFETENSORS_SINGLE_FILE
from termcolor import colored from termcolor import colored
@@ -218,11 +220,6 @@ class WandBLogger:
mode: str = "train", mode: str = "train",
) -> None: ) -> None:
"""Log a small W&B table with sampled images/text and action endpoints.""" """Log a small W&B table with sampled images/text and action endpoints."""
import logging # noqa: PLC0415
import numpy as np # noqa: PLC0415
import torch # noqa: PLC0415
if mode not in {"train", "eval"}: if mode not in {"train", "eval"}:
raise ValueError(mode) raise ValueError(mode)