Skip to content

Commit 013337d

Browse files
fix FT test
1 parent 6c7a72f commit 013337d

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/unitary/with_extras/aqua/test_finetuning.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff 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,

0 commit comments

Comments
 (0)