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: