Fixed wrong builder type.

BerkeleyAutomation · Jun 13, 2024 · 84c08e2 · 84c08e2
1 parent 65c38e7
commit 84c08e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fog_x/dataset.py b/fog_x/dataset.py
@@ -412,7 +412,7 @@ def load_rtx_episodes_local(
         # this is only required if rtx format is used
         import tensorflow_datasets as tfds
 
-        b = tfds.builder(path)
+        b = tfds.builder_from_directory(path)
         self._build_rtx_episodes_from_tfds_builder(
             b,
             split=split,