File tree 2 files changed +2
-1
lines changed
2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -672,6 +672,7 @@ def test_tune_model(
672
672
"pipeline_job"
673
673
].runtime_config .parameter_values
674
674
assert pipeline_arguments ["learning_rate" ] == 0.1
675
+ assert pipeline_arguments ["large_model_reference" ] == "text-bison@001"
675
676
assert (
676
677
call_kwargs ["pipeline_job" ].encryption_spec .kms_key_name
677
678
== _TEST_ENCRYPTION_KEY_NAME
Original file line number Diff line number Diff line change @@ -117,7 +117,7 @@ def _get_model_info(
117
117
118
118
if short_model_id in _SHORT_MODEL_ID_TO_TUNING_PIPELINE_MAP :
119
119
tuning_pipeline_uri = _SHORT_MODEL_ID_TO_TUNING_PIPELINE_MAP [short_model_id ]
120
- tuning_model_id = short_model_id + "-" + publisher_model_res . version_id
120
+ tuning_model_id = publisher_model_template . rsplit ( "/" , 1 )[ - 1 ]
121
121
else :
122
122
tuning_pipeline_uri = None
123
123
tuning_model_id = None
You can’t perform that action at this time.
0 commit comments