fix: LLM - Fixed filter in list_tuned_model_names

sararob · copybara-github · commit 57806fb947e5 · 2023-08-09T20:52:47.000-07:00
PiperOrigin-RevId: 555359431
diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py
@@ -1083,7 +1083,7 @@ def _get_tuned_models_dir_uri(model_id: str) -> str:
 
 def _list_tuned_model_names(model_id: str) -> List[str]:
     tuned_models = aiplatform.Model.list(
-        filter=f'labels.{_TUNING_BASE_MODEL_ID_LABEL_KEY}="{model_id}"',
+        filter=f'labels.{_TUNING_BASE_MODEL_ID_LABEL_KEY}="{model_id.replace("@", "-")}"',
         # TODO(b/275444096): Remove the explicit location once models are deployed to the user's selected location
         location=_TUNED_MODEL_LOCATION,
     )

Original file line number	Diff line number	Diff line change
`@@ -1083,7 +1083,7 @@ def _get_tuned_models_dir_uri(model_id: str) -> str:`
`1083`	`1083`
`1084`	`1084`	`def _list_tuned_model_names(model_id: str) -> List[str]:`
`1085`	`1085`	`tuned_models = aiplatform.Model.list(`
`1086`		`- filter=f'labels.{_TUNING_BASE_MODEL_ID_LABEL_KEY}="{model_id}"',`
	`1086`	`+ filter=f'labels.{_TUNING_BASE_MODEL_ID_LABEL_KEY}="{model_id.replace("@", "-")}"',`
`1087`	`1087`	`# TODO(b/275444096): Remove the explicit location once models are deployed to the user's selected location`
`1088`	`1088`	`location=_TUNED_MODEL_LOCATION,`
`1089`	`1089`	`)`