small fix

2024-03-06 10:21:22 +00:00
parent f95ecd66fc
commit eed24b083a
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/datasets/factory.py
+++ b/lerobot/common/datasets/factory.py
@@ -102,7 +102,7 @@ def make_offline_buffer(
        offline_buffer.set_transform(transform)

    if not overwrite_sampler:
-        index = torch.arange(0, offline_buffer.num_frames, 1)
+        index = torch.arange(0, offline_buffer.num_samples, 1)
        sampler.extend(index)

    return offline_buffer