Publish 'plex_robosuite' OXE datasets, see https://robotics-transformer-x.github.io/.

PiperOrigin-RevId: 628431319

Author: The TensorFlow Datasets Authors

tensorflow_datasets/robotics/__init__.py

@@ -54,6 +54,7 @@
 from tensorflow_datasets.robotics.rtx import NyuDoorOpeningSurprisingEffectiveness
 from tensorflow_datasets.robotics.rtx import NyuFrankaPlayDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx import NyuRotDatasetConvertedExternallyToRlds
+from tensorflow_datasets.robotics.rtx import PlexRobosuite
 from tensorflow_datasets.robotics.rtx import RoboSet
 from tensorflow_datasets.robotics.rtx import Roboturk
 from tensorflow_datasets.robotics.rtx import StanfordHydraDatasetConvertedExternallyToRlds

tensorflow_datasets/robotics/rtx/__init__.py

@@ -54,6 +54,7 @@
 from tensorflow_datasets.robotics.rtx.rtx import NyuDoorOpeningSurprisingEffectiveness
 from tensorflow_datasets.robotics.rtx.rtx import NyuFrankaPlayDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx.rtx import NyuRotDatasetConvertedExternallyToRlds
+from tensorflow_datasets.robotics.rtx.rtx import PlexRobosuite
 from tensorflow_datasets.robotics.rtx.rtx import RoboSet
 from tensorflow_datasets.robotics.rtx.rtx import Roboturk
 from tensorflow_datasets.robotics.rtx.rtx import StanfordHydraDatasetConvertedExternallyToRlds

tensorflow_datasets/robotics/rtx/rtx.py

@@ -1390,3 +1390,23 @@ def get_homepage(self):
 
   def get_relative_dataset_location(self):
     return 'bridge_data_msr/0.0.1'
+
+
+class PlexRobosuite(dataset_importer_builder.DatasetImporterBuilder):
+  """DatasetBuilder for `plex_robosuite` dataset."""
+
+  def get_description(self):
+    return (
+        "A dataset of high-qualty demonstration trajectories for Robosuite's"
+        ' Door, Stack, PickPlaceMilk, PickPlaceBread, PickPlaceCereal, and'
+        ' NutAssemblyRound tasks, 75 demonstrations per each.'
+    )
+
+  def get_citation(self):
+    return 'https://doi.org/10.48550/arXiv.2303.08789'
+
+  def get_homepage(self):
+    return 'https://microsoft.github.io/PLEX/'
+
+  def get_relative_dataset_location(self):
+    return 'plex_robosuite/0.0.1'