From ab5cae6547b864f34676b7c1f845c731cb905138 Mon Sep 17 00:00:00 2001 From: Eugene Mironov Date: Tue, 4 Nov 2025 03:24:45 +0700 Subject: [PATCH] fixup! fixup! fixup! Improve visualization: separate correction plot and fix axis scaling --- src/lerobot/policies/rtc/{debug_handler.py => debug_tracker.py} | 0 src/lerobot/policies/rtc/debug_visualizer.py | 2 +- src/lerobot/policies/rtc/modeling_rtc.py | 2 +- 3 files changed, 2 insertions(+), 2 deletions(-) rename src/lerobot/policies/rtc/{debug_handler.py => debug_tracker.py} (100%) diff --git a/src/lerobot/policies/rtc/debug_handler.py b/src/lerobot/policies/rtc/debug_tracker.py similarity index 100% rename from src/lerobot/policies/rtc/debug_handler.py rename to src/lerobot/policies/rtc/debug_tracker.py diff --git a/src/lerobot/policies/rtc/debug_visualizer.py b/src/lerobot/policies/rtc/debug_visualizer.py index 4c7c218af..6990deee8 100644 --- a/src/lerobot/policies/rtc/debug_visualizer.py +++ b/src/lerobot/policies/rtc/debug_visualizer.py @@ -21,7 +21,7 @@ import torch from matplotlib.figure import Figure from torch import Tensor -from lerobot.policies.rtc.debug_handler import Tracker +from lerobot.policies.rtc.debug_tracker import Tracker class RTCDebugVisualizer: diff --git a/src/lerobot/policies/rtc/modeling_rtc.py b/src/lerobot/policies/rtc/modeling_rtc.py index b6feed47a..9d231ec79 100644 --- a/src/lerobot/policies/rtc/modeling_rtc.py +++ b/src/lerobot/policies/rtc/modeling_rtc.py @@ -29,7 +29,7 @@ from torch import Tensor from lerobot.configs.types import RTCAttentionSchedule from lerobot.policies.rtc.configuration_rtc import RTCConfig -from lerobot.policies.rtc.debug_handler import Tracker +from lerobot.policies.rtc.debug_tracker import Tracker logger = logging.getLogger(__name__)