From b804c9adaf450613fc245d9c754c7914bc2a1496 Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Wed, 28 Jan 2026 11:09:26 -0800
Subject: [PATCH] Expose `FlatMapTransform` in grain `__init__.py`

PiperOrigin-RevId: 862313774
---
 grain/python/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/grain/python/__init__.py b/grain/python/__init__.py
index 0ac9d7d1f..1577dcfd4 100644
--- a/grain/python/__init__.py
+++ b/grain/python/__init__.py
@@ -37,6 +37,7 @@
     Batch,
     Filter as FilterTransform,
     Map as MapTransform,
+    FlatMap as FlatMapTransform,
     MapWithIndex as MapWithIndexTransform,
     RandomMap as RandomMapTransform,
     Transformation,