diff --git a/src/lerobot/policies/rtc/modeling_rtc.py b/src/lerobot/policies/rtc/modeling_rtc.py index 041ef7e1e..17db3762d 100644 --- a/src/lerobot/policies/rtc/modeling_rtc.py +++ b/src/lerobot/policies/rtc/modeling_rtc.py @@ -53,7 +53,7 @@ class RTCProcessor: ) # ====================== Tracker Proxy Methods ====================== - def track_debug( + def track( self, time: float | Tensor, x_t: Tensor | None = None, @@ -261,7 +261,7 @@ class RTCProcessor: err = err.squeeze(0) # Record debug information (all params except x_t which is recorded externally) - self.track_debug( + self.track( time=time, v_t=v_t, x1_t=x1_t, diff --git a/src/lerobot/policies/smolvla/modeling_smolvla.py b/src/lerobot/policies/smolvla/modeling_smolvla.py index a9cdbf7a0..0842062d3 100644 --- a/src/lerobot/policies/smolvla/modeling_smolvla.py +++ b/src/lerobot/policies/smolvla/modeling_smolvla.py @@ -831,7 +831,7 @@ class VLAFlowMatching(nn.Module): # Record x_t after Euler step (other params are recorded in rtc_processor.denoise_step) if self._rtc_enabled() and correction is not None: - self.rtc_processor.track_debug(time=time, x_t=x_t) + self.rtc_processor.track(time=time, x_t=x_t) # Visualize x_t using plot_waypoints - accumulate all denoise steps # Use provided axes or create new ones