diff --git a/src/lerobot/utils/visualization_utils.py b/src/lerobot/utils/visualization_utils.py index 4a55165495..f0f9aebb74 100644 --- a/src/lerobot/utils/visualization_utils.py +++ b/src/lerobot/utils/visualization_utils.py @@ -38,9 +38,9 @@ def log_rerun_data(observation: dict[str | Any], action: dict[str | Any]): rr.log(f"observation.{obs}_{i}", rr.Scalar(float(v))) else: rr.log(f"observation.{obs}", rr.Image(val), static=True) - for act, val in action.items(): - if isinstance(val, float): - rr.log(f"action.{act}", rr.Scalar(val)) - elif isinstance(val, np.ndarray): - for i, v in enumerate(val): - rr.log(f"action.{act}_{i}", rr.Scalar(float(v))) + for act, val in action.items(): + if isinstance(val, float): + rr.log(f"action.{act}", rr.Scalar(val)) + elif isinstance(val, np.ndarray): + for i, v in enumerate(val): + rr.log(f"action.{act}_{i}", rr.Scalar(float(v)))