From 08907eee9ede83d7b2b0887679d2fa52952a53cd Mon Sep 17 00:00:00 2001 From: Grain Team Date: Wed, 15 Oct 2025 07:38:42 -0700 Subject: [PATCH] Internal PiperOrigin-RevId: 819749585 --- grain/python/experimental.py | 1 + 1 file changed, 1 insertion(+) diff --git a/grain/python/experimental.py b/grain/python/experimental.py index c213ae4d3..0da068eca 100644 --- a/grain/python/experimental.py +++ b/grain/python/experimental.py @@ -43,6 +43,7 @@ from grain._src.python.dataset.transformations.map import RngPool from grain._src.python.dataset.transformations.mix import ConcatenateMapDataset from grain._src.python.dataset.transformations.packing import FirstFitPackIterDataset +from grain._src.python.dataset.transformations.packing import PackingDatasetIterator from grain._src.python.dataset.transformations.packing_concat_then_split import ( BOSHandling, ConcatThenSplitIterDataset,