Merge pull request #16 from utn-blei/main

fixed attribute error.

fog_x/dataset.py

@@ -412,15 +412,18 @@ 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,
             additional_metadata=additional_metadata,
         )
 
     def _build_rtx_episodes_from_tfds_builder(
+        self,
         builder,
+        split: str = "all",
+        additional_metadata: Optional[Dict[str, Any]] = dict(),
     ):
         """
         construct the dataset from the tfds builder