Rename track_debug method to track

Simplify the method name from track_debug to just track for better
readability and consistency. The method already has clear documentation
about its debug tracking purpose.

Changes:
- Rename RTCProcessor.track_debug() to track()
- Update all call sites in modeling_smolvla.py and modeling_rtc.py

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Alexander Soare <alexander.soare159@gmail.com>
Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Eugene Mironov
2025-11-03 18:20:41 +07:00
parent 05189361b6
commit d20ef2e46e
2 changed files with 3 additions and 3 deletions

View File

@@ -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,

View File

@@ -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