Move RLDS datasets into a separate subfolder

This will allow to more easily import only the datasets builders or the generic helper code.

PiperOrigin-RevId: 636085996

Author: tomvdw

tensorflow_datasets/rlds/__init__.py

@@ -14,5 +14,7 @@
 # limitations under the License.
 
 """Datasets generated with RLDS."""
-from tensorflow_datasets.rlds.locomotion.locomotion import Locomotion
-from tensorflow_datasets.rlds.robosuite_panda_pick_place_can.robosuite_panda_pick_place_can import RobosuitePandaPickPlaceCan
+
+from tensorflow_datasets.rlds import datasets
+from tensorflow_datasets.rlds import envlogger_reader
+from tensorflow_datasets.rlds import rlds_base

tensorflow_datasets/rlds/datasets/__init__.py

@@ -0,0 +1,19 @@
+# coding=utf-8
+# Copyright 2024 The TensorFlow Datasets Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Datasets generated with RLDS."""
+
+from tensorflow_datasets.rlds.datasets.locomotion.locomotion import Locomotion
+from tensorflow_datasets.rlds.datasets.robosuite_panda_pick_place_can.robosuite_panda_pick_place_can import RobosuitePandaPickPlaceCan

tensorflow_datasets/rlds/locomotion/__init__.py renamed to tensorflow_datasets/rlds/datasets/locomotion/__init__.py

@@ -15,4 +15,4 @@
 
 """locomotion dataset."""
 
-from tensorflow_datasets.rlds.locomotion.locomotion import Locomotion
+from tensorflow_datasets.rlds.datasets.locomotion.locomotion import Locomotion