File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff line change @@ -90,6 +90,10 @@ def test_create_fine_tuning(
9090 ft_source .compartment_id = self .SERVICE_COMPARTMENT_ID
9191 ft_source .display_name = "test_ft_source_model"
9292 ft_source .custom_metadata_list = custom_metadata_list
93+ ft_source .freeform_tags = {
94+ "license" : "Some license text" ,
95+ "aqua_custom_base_model" : "base_model_info" ,
96+ }
9397 mock_get_source .return_value = ft_source
9498
9599 mock_mvs_create .return_value = ("test_experiment_id" , "test_experiment_name" )
@@ -177,6 +181,9 @@ def test_create_fine_tuning(
177181 "finetuning_experiment_id" : f"{ mock_mvs_create .return_value [0 ]} " ,
178182 "finetuning_job_id" : f"{ mock_job_id .return_value } " ,
179183 "finetuning_source" : f"{ ft_source .id } " ,
184+ "ready_to_fine_tune" : "false" ,
185+ "OCI_AQUA" : "" ,
186+ "aqua_fine_tuned_model" : f"{ ft_source .id } #{ ft_source .display_name } " ,
180187 },
181188 ** ft_model_freeform_tags ,
182189 ** ft_model_defined_tags ,
You can’t perform that action at this time.
0 commit comments