add image embeddings to __all__

apache · Jun 6, 2024 · bb152c7 · bb152c7
1 parent 602045d
commit bb152c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdks/python/apache_beam/ml/transforms/embeddings/vertex_ai.py b/sdks/python/apache_beam/ml/transforms/embeddings/vertex_ai.py
@@ -40,7 +40,7 @@
 from vertexai.vision_models import Image
 from vertexai.vision_models import MultiModalEmbeddingModel
 
-__all__ = ["VertexAITextEmbeddings"]
+__all__ = ["VertexAITextEmbeddings", "VertexAIImageEmbeddings"]
 
 DEFAULT_TASK_TYPE = "RETRIEVAL_DOCUMENT"
 # TODO: https://github.com/apache/beam/issues/29356