chore: Add test for model_version_id when using predict(..., use_raw_predict=True)

vertex-sdk-bot · copybara-github · commit c23a8bdfcc77 · 2022-12-19T17:04:05.000-08:00
PiperOrigin-RevId: 496524822
diff --git a/tests/system/aiplatform/test_model_interactions.py b/tests/system/aiplatform/test_model_interactions.py
@@ -52,6 +52,10 @@ def test_prediction(self):
             prediction_with_raw_predict.model_resource_name
             == prediction_response.model_resource_name
         )
+        assert (
+            prediction_with_raw_predict.model_version_id
+            == prediction_response.model_version_id
+        )
 
         # test raw_predict
         raw_prediction_response = self.endpoint.raw_predict(
