diff --git a/lerobot/common/datasets/profile_streaming_dataset.py b/lerobot/common/datasets/profile_streaming_dataset.py index 99d0cc8f..a9d02c4c 100644 --- a/lerobot/common/datasets/profile_streaming_dataset.py +++ b/lerobot/common/datasets/profile_streaming_dataset.py @@ -163,7 +163,7 @@ def _profile_iteration(dataset, num_samples, stats_file_path): # Add functions to profile profiler.add_function(dataset.__iter__) profiler.add_function(dataset.make_frame) - profiler.add_function(dataset._make_iterable_dataset) + profiler.add_function(dataset._make_backtrackable_dataset) # Profile the iteration