File tree 2 files changed +9
-2
lines changed
2 files changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -2385,7 +2385,8 @@ def test_tune_text_embedding_model(
2385
2385
model = language_models_module .TextEmbeddingModel .from_pretrained (
2386
2386
base_model_version_id
2387
2387
)
2388
- tuning_job = model .tune_model (** tune_args )
2388
+ tuning_result = model .tune_model (** tune_args )
2389
+
2389
2390
call_kwargs = mock_pipeline_service_create .call_args [1 ]
2390
2391
pipeline_arguments = dict (
2391
2392
call_kwargs ["pipeline_job" ].runtime_config .parameter_values
@@ -2402,7 +2403,8 @@ def test_tune_text_embedding_model(
2402
2403
assert pipeline_arguments == expected_pipeline_args
2403
2404
2404
2405
# Testing the tuned model
2405
- tuned_model = tuning_job .deploy_tuned_model ()
2406
+ assert tuning_result .pipeline_job_name .startswith ("sample-test-pipeline" )
2407
+ tuned_model = tuning_result .deploy_tuned_model ()
2406
2408
assert (
2407
2409
tuned_model ._endpoint_name
2408
2410
== test_constants .EndpointConstants ._TEST_ENDPOINT_NAME
Original file line number Diff line number Diff line change @@ -3932,6 +3932,11 @@ def _get_tuned_model_name(self) -> str:
3932
3932
self ._tuned_model_name = vertex_model_name
3933
3933
return vertex_model_name
3934
3934
3935
+ @property
3936
+ def pipeline_job_name (self ) -> str :
3937
+ """Pipeline job name."""
3938
+ return self ._job .name
3939
+
3935
3940
@property
3936
3941
def _status (self ) -> Optional [aiplatform_types .pipeline_state .PipelineState ]:
3937
3942
"""Job status."""
You can’t perform that action at this time.
0 commit comments