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

PiperOrigin-RevId: 632204349

Author: The TensorFlow Datasets Authors

tensorflow_datasets/robotics/__init__.py

@@ -57,6 +57,7 @@
 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 SpocRobot
 from tensorflow_datasets.robotics.rtx import StanfordHydraDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx import StanfordKukaMultimodalDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx import StanfordMaskVitConvertedExternallyToRlds

tensorflow_datasets/robotics/rtx/__init__.py

@@ -57,6 +57,7 @@
 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 SpocRobot
 from tensorflow_datasets.robotics.rtx.rtx import StanfordHydraDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx.rtx import StanfordKukaMultimodalDatasetConvertedExternallyToRlds
 from tensorflow_datasets.robotics.rtx.rtx import StanfordMaskVitConvertedExternallyToRlds

tensorflow_datasets/robotics/rtx/rtx.py

@@ -1410,3 +1410,25 @@ def get_homepage(self):
 
   def get_relative_dataset_location(self):
     return 'plex_robosuite/0.0.1'
+
+
+class SpocRobot(dataset_importer_builder.DatasetImporterBuilder):
+  """DatasetBuilder for `spoc_robot` dataset."""
+
+  def get_description(self):
+    return ''
+
+  def get_citation(self):
+    return r"""@article{spoc2023,
+    author    = {Kiana Ehsani, Tanmay Gupta, Rose Hendrix, Jordi Salvador, Luca Weihs, Kuo-Hao Zeng, Kunal Pratap Singh, Yejin Kim, Winson Han, Alvaro Herrasti, Ranjay Krishna, Dustin Schwenk, Eli VanderBilt, Aniruddha Kembhavi},
+    title     = {Imitating Shortest Paths in Simulation Enables Effective Navigation and Manipulation in the Real World},
+    journal   = {arXiv},
+    year      = {2023},
+    eprint    = {2312.02976},
+}"""
+
+  def get_homepage(self):
+    return 'https://spoc-robot.github.io/'
+
+  def get_relative_dataset_location(self):
+    return 'spoc/0.0.1'