diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py index e186b140ba..6c62ccef42 100644 --- a/google/cloud/aiplatform/models.py +++ b/google/cloud/aiplatform/models.py @@ -4150,7 +4150,7 @@ def raw_predict( url = f"https://{endpoint_override}/v1/projects/{self.project}/locations/{self.location}/endpoints/{self.name}:rawPredict" return self._authorized_session.post( url=url, - body=body, + data=body, headers=headers, )