Add BaseBeamComponent to tfx.v1.types.

PiperOrigin-RevId: 594163722
stevenschaerer · Dec 28, 2023 · c1797a5 · c1797a5
1 parent da0d5a3
commit c1797a5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tfx/v1/types/__init__.py b/tfx/v1/types/__init__.py
@@ -16,6 +16,7 @@
 # Base types are meant for type hinting only. Abstrain from directly inheriting
 # base types for your own classes.
 # pylint: disable=g-importing-member
+from tfx.dsl.components.base.base_beam_component import BaseBeamComponent
 from tfx.dsl.components.base.base_component import BaseComponent
 from tfx.dsl.components.base.base_node import BaseNode
 from tfx.types.channel import BaseChannel