Add api docs for tft.beam

smokestacklightnin · smokestacklightnin · commit 11ba8c94ae9d · 2025-05-13T17:42:36.000-07:00
diff --git a/docs/api_docs/python/tft-beam.md b/docs/api_docs/python/tft-beam.md
@@ -0,0 +1,3 @@
+# TensorFlow Transform `tft.beam` Module
+
+::: tensorflow_transform.beam
diff --git a/mkdocs.yml b/mkdocs.yml
@@ -128,3 +128,4 @@ nav:
       - tft: api_docs/python/tft.md
       - tft.coders: api_docs/python/tft-coders.md
       - tft.experimental 🧪: api_docs/python/tft-experimental.md
+      - tft.beam: api_docs/python/tft-beam.md
diff --git a/tensorflow_transform/beam/__init__.py b/tensorflow_transform/beam/__init__.py
@@ -37,3 +37,15 @@
   import tensorflow_io as _  # pytype: disable=import-error # pylint: disable=g-import-not-at-top
 except ModuleNotFoundError:
   pass
+
+__all__ = [
+    "AnalyzeAndTransformDataset",
+    "AnalyzeDataset",
+    "AnalyzeDatasetWithCache",
+    "Context",
+    "EncodeTransformedDataset",
+    "ReadTransformFn",
+    "TransformDataset",
+    "WriteMetadata",
+    "WriteTransformFn",
+]

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	+# TensorFlow Transform `tft.beam` Module
	`2`	`+`
	`3`	`+::: tensorflow_transform.beam`