Fix dataset repeat

kpertsch · web-flow · commit e4580662a60e · 2025-09-07T10:03:31.000-07:00
diff --git a/src/openpi/training/droid_rlds_dataset.py b/src/openpi/training/droid_rlds_dataset.py
@@ -59,7 +59,7 @@ def __init__(
         )
 
         # # Repeat dataset so we never run out of data.
-        # dataset = dataset.repeat()
+        dataset = dataset.repeat()
 
         # Load the filter dictionary if provided.
         # The filter dictionary is a JSON file that maps episode keys to ranges of frames to sample

Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,7 @@ def __init__(`
`59`	`59`	`)`
`60`	`60`
`61`	`61`	`# # Repeat dataset so we never run out of data.`
`62`		`- # dataset = dataset.repeat()`
	`62`	`+ dataset = dataset.repeat()`
`63`	`63`
`64`	`64`	`# Load the filter dictionary if provided.`
`65`	`65`	`# The filter dictionary is a JSON file that maps episode keys to ranges of frames to sample`