Remove lazy_dataset after it has been deprecated for a while.

PiperOrigin-RevId: 713389900

grain/BUILD

@@ -49,23 +49,12 @@ py_library(
     srcs_version = "PY3",
     # Implicit build flag
     deps = [
-        ":python_lazy_dataset",  # build_cleaner: keep
         "//grain/_src/core:transforms",  # build_cleaner: keep
         "//grain/_src/python/dataset:visualize",  # build_cleaner: keep
-        "//grain/_src/python/dataset/transformations:interleave",  # build_cleaner: keep
-        "//grain/_src/python/experimental/example_packing:packing",  # build_cleaner: keep
-    ],
-)
-
-py_library(
-    name = "python_lazy_dataset",
-    srcs = ["python_lazy_dataset.py"],
-    srcs_version = "PY3",
-    # Implicit build flag
-    deps = [
-        "//grain/_src/python/dataset",  # build_cleaner: keep
         "//grain/_src/python/dataset/transformations:flatmap",  # build_cleaner: keep
+        "//grain/_src/python/dataset/transformations:interleave",  # build_cleaner: keep
         "//grain/_src/python/dataset/transformations:packing",  # build_cleaner: keep
         "//grain/_src/python/dataset/transformations:zip",  # build_cleaner: keep
+        "//grain/_src/python/experimental/example_packing:packing",  # build_cleaner: keep
     ],
 )

grain/_src/python/dataset/dataset.py

@@ -11,7 +11,7 @@
 # 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.
-"""LazyDataset base classes.
+"""Dataset base classes.
 
 There are 3 main classes:
 - `MapDataset` define a dataset that supports efficient random access. It

grain/python_experimental.py

@@ -20,16 +20,6 @@
 # pylint: disable=wildcard-import
 # pylint: disable=g-import-not-at-top
 
-from etils import epy
-
-# `lazy_dataset` module is deprecated. It displays the deprecation message upon
-# import. We make the import lazy to only display the message when the module
-# is actually used.
-with epy.lazy_imports():
-  from grain import python_lazy_dataset as lazy_dataset
-
-del epy
-
 from ._src.python.dataset.base import (
     DatasetOptions,
     ExecutionTrackingMode,