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

This commit is contained in:
Pepijn
2026-07-15 13:41:39 +02:00
parent 94dc85b443
commit 87585195e6
+2 -5
View File
@@ -19,6 +19,8 @@ import re
from glob import glob
from pathlib import Path
import numpy as np
import torch
from huggingface_hub.constants import SAFETENSORS_SINGLE_FILE
from termcolor import colored
@@ -218,11 +220,6 @@ class WandBLogger:
mode: str = "train",
) -> None:
"""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"}:
raise ValueError(mode)