fix: create FV embedding dimensions sample - dimensions should be an int

vertex-sdk-bot · copybara-github · commit 2aa221ec994f · 2024-07-10T11:28:03.000-07:00
PiperOrigin-RevId: 651092279
diff --git a/samples/model-builder/feature_store/create_feature_view_from_bq_source_with_embedding_management.py b/samples/model-builder/feature_store/create_feature_view_from_bq_source_with_embedding_management.py
@@ -27,7 +27,7 @@ def create_feature_view_from_bq_source_with_embedding_management(
     bq_table_uri: str,
     entity_id_columns: List[str],
     embedding_column: str,
-    embedding_dimensions: str,
+    embedding_dimensions: int,
 ):
     aiplatform.init(project=project, location=location)
 
