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__)