From 65c38e7d7eb0001ff41f54f74a84126e004f2879 Mon Sep 17 00:00:00 2001 From: Yannik Blei Date: Thu, 13 Jun 2024 12:02:02 +0200 Subject: [PATCH 1/2] fixed attribute error. --- fog_x/dataset.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/fog_x/dataset.py b/fog_x/dataset.py index ed0c98c..cd59771 100644 --- a/fog_x/dataset.py +++ b/fog_x/dataset.py @@ -420,7 +420,10 @@ def load_rtx_episodes_local( ) 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 From 84c08e222424f4c5dc002fdd6298f4a22c90ff48 Mon Sep 17 00:00:00 2001 From: Yannik Blei Date: Thu, 13 Jun 2024 15:24:46 +0200 Subject: [PATCH 2/2] Fixed wrong builder type. --- fog_x/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fog_x/dataset.py b/fog_x/dataset.py index cd59771..f20d343 100644 --- 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,