File tree 2 files changed +4
-4
lines changed
2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -225,6 +225,7 @@ def create_model_catalog(
225
225
model_taxonomy_metadata : Union [ModelTaxonomyMetadata , Dict ],
226
226
compartment_id : str ,
227
227
project_id : str ,
228
+ defined_tags : Dict = None ,
228
229
** kwargs ,
229
230
) -> DataScienceModel :
230
231
model = (
@@ -237,7 +238,7 @@ def create_model_catalog(
237
238
.with_custom_metadata_list (model_custom_metadata )
238
239
.with_defined_metadata_list (model_taxonomy_metadata )
239
240
.with_provenance_metadata (ModelProvenanceMetadata (training_id = UNKNOWN ))
240
- # TODO: decide what parameters will be needed
241
+ . with_defined_tags ( ** ( defined_tags or {})) # Create defined tags when a model is created.
241
242
.create (
242
243
** kwargs ,
243
244
)
Original file line number Diff line number Diff line change @@ -263,6 +263,7 @@ def create(
263
263
compartment_id = target_compartment ,
264
264
project_id = target_project ,
265
265
model_by_reference = True ,
266
+ defined_tags = create_fine_tuning_details .defined_tags
266
267
)
267
268
268
269
ft_job_freeform_tags = {
@@ -382,14 +383,12 @@ def create(
382
383
Tags .AQUA_FINE_TUNED_MODEL_TAG : f"{ source .id } #{ source .display_name } " ,
383
384
** (create_fine_tuning_details .freeform_tags or {}),
384
385
}
385
- model_defined_tags = create_fine_tuning_details .defined_tags or {}
386
386
387
387
self .update_model (
388
388
model_id = ft_model .id ,
389
389
update_model_details = UpdateModelDetails (
390
390
custom_metadata_list = updated_custom_metadata_list ,
391
391
freeform_tags = model_freeform_tags ,
392
- defined_tags = model_defined_tags ,
393
392
),
394
393
)
395
394
logger .debug (
@@ -490,7 +489,7 @@ def create(
490
489
"finetuning_source" : source .id ,
491
490
"finetuning_experiment_id" : experiment_model_version_set_id ,
492
491
** model_freeform_tags ,
493
- ** model_defined_tags ,
492
+ ** ( create_fine_tuning_details . defined_tags or {}) ,
494
493
},
495
494
parameters = ft_parameters ,
496
495
)
You can’t perform that action at this time.
0 commit comments