Skip to content

Commit 64fcfda

Browse files
sararobcopybara-github
authored andcommitted
fix: Remove lazy loading of vertexai.types module to enable from vertexai.types import TypeName
PiperOrigin-RevId: 834811652
1 parent 186e6d8 commit 64fcfda

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

vertexai/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@
1818
import sys
1919

2020
from google.cloud.aiplatform import version as aiplatform_version
21+
from vertexai._genai import types
22+
23+
# sys.modules[__name__ + ".types"] = types
2124

2225
__version__ = aiplatform_version.__version__
2326

0 commit comments

Comments
 (0)