From 9a9d70b27bf2832e59858aaaab6de34e2cc6036f Mon Sep 17 00:00:00 2001 From: Grain Team Date: Fri, 13 Feb 2026 13:50:40 -0800 Subject: [PATCH] Internal PiperOrigin-RevId: 869874852 --- grain/python/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/grain/python/__init__.py b/grain/python/__init__.py index 0ac9d7d1f..c26e7cbb9 100644 --- a/grain/python/__init__.py +++ b/grain/python/__init__.py @@ -58,6 +58,7 @@ from grain._src.python.dataset.base import ( DatasetSelectionMap, RandomAccessDataSource, + SupportsBatchedReadRandomAccessDataSource, ) from grain._src.python.dataset.dataset import ( MapDataset,