Expose VarLenTensorValue in top-level __init__.py

PiperOrigin-RevId: 442041497
zywind · Apr 15, 2022 · 4bfc817 · 4bfc817
1 parent 2454289
commit 4bfc817
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow_model_analysis/__init__.py b/tensorflow_model_analysis/__init__.py
@@ -134,6 +134,7 @@
   from tensorflow_model_analysis.types import TensorType
   from tensorflow_model_analysis.types import TensorTypeMaybeDict
   from tensorflow_model_analysis.types import TensorValue
+  from tensorflow_model_analysis.types import VarLenTensorValue
 
   # Import VERSION as __version__ for compatibility with other TFX components.
   from tensorflow_model_analysis.version import VERSION as __version__