From 4bdbf2f6e0082989f4424d0a103944431a1b8429 Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Fri, 14 Mar 2025 16:59:31 +0100
Subject: [PATCH] update comment

---
 lerobot/common/datasets/sampler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lerobot/common/datasets/sampler.py b/lerobot/common/datasets/sampler.py
index f26fba53..ddd1176e 100644
--- a/lerobot/common/datasets/sampler.py
+++ b/lerobot/common/datasets/sampler.py
@@ -140,8 +140,8 @@ class PrioritizedSampler(Sampler[int]):
         Args:
             data_len: Total number of samples in the dataset.
             alpha: Exponent for priority scaling. Default is 0.6.
+            beta: Used in important sampling (IS). Default is 0.4,
             eps: Small constant to avoid zero priorities.
-            replacement: Whether to sample with replacement.
             num_samples_per_epoch: Number of samples per epoch (default is data_len).
         """
         self.data_len = data_len