chore: use keyword arguments in raw_predict implementation

PiperOrigin-RevId: 506090900
googleapis · Jan 31, 2023 · 7ab6e0b · 7ab6e0b
1 parent d83bc79
commit 7ab6e0b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py
@@ -1577,7 +1577,9 @@ def raw_predict(
             )
             self.raw_predict_request_url = f"https://{self.location}-{constants.base.API_BASE_PATH}/v1/projects/{self.project}/locations/{self.location}/endpoints/{self.name}:rawPredict"
 
-        return self.authorized_session.post(self.raw_predict_request_url, body, headers)
+        return self.authorized_session.post(
+            url=self.raw_predict_request_url, data=body, headers=headers
+        )
 
     def explain(
         self,

diff --git a/tests/unit/aiplatform/test_models.py b/tests/unit/aiplatform/test_models.py
@@ -61,6 +61,10 @@
 
 from google.protobuf import field_mask_pb2, timestamp_pb2
 
+from test_endpoints import (  # noqa: F401
+    create_endpoint_mock,
+)
+
 _TEST_PROJECT = "test-project"
 _TEST_PROJECT_2 = "test-project-2"
 _TEST_LOCATION = "us-central1"
@@ -2782,5 +2786,7 @@ def test_raw_predict(self, raw_predict_mock):
         test_endpoint = models.Endpoint(_TEST_ID)
         test_endpoint.raw_predict(_TEST_RAW_PREDICT_DATA, _TEST_RAW_PREDICT_HEADER)
         raw_predict_mock.assert_called_once_with(
-            _TEST_RAW_PREDICT_URL, _TEST_RAW_PREDICT_DATA, _TEST_RAW_PREDICT_HEADER
+            url=_TEST_RAW_PREDICT_URL,
+            data=_TEST_RAW_PREDICT_DATA,
+            headers=_TEST_RAW_PREDICT_HEADER,
         )