fix: Make statistics Optional for TextEmbedding.

sasha-gitg · copybara-github · commit 7eaa1d4216f0 · 2023-08-21T16:35:53.000-07:00
PiperOrigin-RevId: 558933105
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
