fix: Make statistics Optional for TextEmbedding.

PiperOrigin-RevId: 558933105
googleapis · Aug 21, 2023 · 7eaa1d4 · 7eaa1d4
1 parent ff5cfa1
commit 7eaa1d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py
@@ -1133,8 +1133,8 @@ class TextEmbedding:
     """Text embedding vector and statistics."""
 
     values: List[float]
-    statistics: TextEmbeddingStatistics
-    _prediction_response: aiplatform.models.Prediction = None
+    statistics: Optional[TextEmbeddingStatistics] = None
+    _prediction_response: Optional[aiplatform.models.Prediction] = None
 
 
 @dataclasses.dataclass