googleapis
diff --git a/‎.kokoro/samples/lint/presubmit.cfg‎
Lines changed: 0 additions & 6 deletions b/‎.kokoro/samples/lint/presubmit.cfg‎
Lines changed: 0 additions & 6 deletions
diff --git a/‎.kokoro/samples/python3.10/presubmit.cfg‎
Lines changed: 0 additions & 6 deletions b/‎.kokoro/samples/python3.10/presubmit.cfg‎
Lines changed: 0 additions & 6 deletions
diff --git a/‎tests/unit/aiplatform/test_multimodal_datasets.py‎
Lines changed: 3 additions & 2 deletions b/‎tests/unit/aiplatform/test_multimodal_datasets.py‎
Lines changed: 3 additions & 2 deletions
@@ -1217,8 +1217,9 @@ def test_gemini_example_from_prompt(self):
         )
         gemini_example = ummd.GeminiExample.from_prompt(prompt)
         assert (
-            gemini_example.model
-            == "projects/test-project/locations/us-central1/publishers/google/models/gemini-1.5-pro-002"
+            gemini_example.model.endswith(
+                "locations/us-central1/publishers/google/models/gemini-1.5-pro-002"
+            )
         )
         assert str(gemini_example.contents) == str(
             [
Original file line number	Diff line number	Diff line change
`@@ -1217,8 +1217,9 @@ def test_gemini_example_from_prompt(self):`
`1217`	`1217`	`)`
`1218`	`1218`	`gemini_example = ummd.GeminiExample.from_prompt(prompt)`
`1219`	`1219`	`assert (`
`1220`		`- gemini_example.model`
`1221`		`- == "projects/test-project/locations/us-central1/publishers/google/models/gemini-1.5-pro-002"`
	`1220`	`+ gemini_example.model.endswith(`
	`1221`	`+ "locations/us-central1/publishers/google/models/gemini-1.5-pro-002"`
	`1222`	`+ )`
`1222`	`1223`	`)`
`1223`	`1224`	`assert str(gemini_example.contents) == str(`
`1224`	`1225`	`[`