admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow dynamic range timing counters

Browse Source
This commit is contained in:
Pepijn
2026-06-19 09:43:57 +02:00
parent 28c3e095bf
commit 03fc5e3ea9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/episode_video_streaming.py
+2 -2
View File
@@ -113,7 +113,7 @@ class ThreadLocalRangeFetcher:
def _record_timing(self, **kwargs: float) -> None: def _record_timing(self, **kwargs: float) -> None:
with self._timing_lock: with self._timing_lock:
for key, value in kwargs.items(): 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]: def timing_summary(self) -> dict[str, float]:
with self._timing_lock: with self._timing_lock:
@@ -419,7 +419,7 @@ class NativeHTTPRangeFetcher:
def _record_timing(self, **kwargs: float) -> None: def _record_timing(self, **kwargs: float) -> None:
with self._timing_lock: with self._timing_lock:
for key, value in kwargs.items(): 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]: def timing_summary(self) -> dict[str, float]:
with self._timing_lock: with self._timing_lock: