diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py index 893770bffa..9b46420c32 100644 --- a/google/cloud/aiplatform/models.py +++ b/google/cloud/aiplatform/models.py @@ -147,7 +147,7 @@ class Prediction(NamedTuple): of elements as instances to be explained. Default is None. """ - predictions: List[Dict[str, Any]] + predictions: List[Any] deployed_model_id: str model_version_id: Optional[str] = None model_resource_name: Optional[str] = None