From 03fc5e3ea9018877d5114c97996f7703371741c2 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Fri, 19 Jun 2026 09:43:57 +0200 Subject: [PATCH] Allow dynamic range timing counters --- src/lerobot/datasets/episode_video_streaming.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lerobot/datasets/episode_video_streaming.py b/src/lerobot/datasets/episode_video_streaming.py index 4f4b8a47f..22fea4116 100644 --- a/src/lerobot/datasets/episode_video_streaming.py +++ b/src/lerobot/datasets/episode_video_streaming.py @@ -113,7 +113,7 @@ class ThreadLocalRangeFetcher: def _record_timing(self, **kwargs: float) -> None: with self._timing_lock: for key, value in kwargs.items(): - self._timing_totals[key] += value + self._timing_totals[key] = self._timing_totals.get(key, 0.0) + value def timing_summary(self) -> dict[str, float]: with self._timing_lock: @@ -419,7 +419,7 @@ class NativeHTTPRangeFetcher: def _record_timing(self, **kwargs: float) -> None: with self._timing_lock: for key, value in kwargs.items(): - self._timing_totals[key] += value + self._timing_totals[key] = self._timing_totals.get(key, 0.0) + value def timing_summary(self) -> dict[str, float]: with self._timing_lock: