@@ -260,6 +260,10 @@ def create(
260260 ** create_aqua_evaluation_details .model_parameters ,
261261 )
262262
263+ evaluation_model_defined_tags = (
264+ create_aqua_evaluation_details .defined_tags or {}
265+ )
266+
263267 target_compartment = (
264268 create_aqua_evaluation_details .compartment_id or COMPARTMENT_OCID
265269 )
@@ -311,9 +315,7 @@ def create(
311315 create_aqua_evaluation_details .experiment_description
312316 )
313317 .with_freeform_tags (** evaluation_mvs_freeform_tags )
314- .with_defined_tags (
315- ** (create_aqua_evaluation_details .defined_tags or {})
316- )
318+ .with_defined_tags (** evaluation_model_defined_tags )
317319 # TODO: decide what parameters will be needed
318320 .create (** kwargs )
319321 )
@@ -358,6 +360,7 @@ def create(
358360 .with_custom_metadata_list (evaluation_model_custom_metadata )
359361 .with_defined_metadata_list (evaluation_model_taxonomy_metadata )
360362 .with_provenance_metadata (ModelProvenanceMetadata (training_id = UNKNOWN ))
363+ .with_defined_tags (** evaluation_model_defined_tags )
361364 # TODO uncomment this once the evaluation container will get the updated version of the ADS
362365 # .with_input_schema(create_aqua_evaluation_details.to_dict())
363366 # TODO: decide what parameters will be needed
@@ -390,7 +393,7 @@ def create(
390393 .with_shape_name (create_aqua_evaluation_details .shape_name )
391394 .with_block_storage_size (create_aqua_evaluation_details .block_storage_size )
392395 .with_freeform_tag (** evaluation_job_freeform_tags )
393- .with_defined_tag (** ( create_aqua_evaluation_details . defined_tags or {}) )
396+ .with_defined_tag (** evaluation_model_defined_tags )
394397 )
395398 if (
396399 create_aqua_evaluation_details .memory_in_gbs
@@ -429,15 +432,17 @@ def create(
429432 metrics = create_aqua_evaluation_details .metrics ,
430433 inference_configuration = eval_inference_configuration or {},
431434 )
432- ).create (** kwargs ) ## TODO: decide what parameters will be needed
435+ ).create (
436+ ** kwargs
437+ ) ## TODO: decide what parameters will be needed
433438 logger .debug (
434439 f"Successfully created evaluation job { evaluation_job .id } for { create_aqua_evaluation_details .evaluation_source_id } ."
435440 )
436441
437442 evaluation_job_run = evaluation_job .run (
438443 name = evaluation_model .display_name ,
439444 freeform_tags = evaluation_job_freeform_tags ,
440- defined_tags = ( create_aqua_evaluation_details . defined_tags or {}) ,
445+ defined_tags = evaluation_model_defined_tags ,
441446 wait = False ,
442447 )
443448 logger .debug (
@@ -461,16 +466,12 @@ def create(
461466 Tags .AQUA_EVALUATION : Tags .AQUA_EVALUATION ,
462467 ** (create_aqua_evaluation_details .freeform_tags or {}),
463468 }
464- evaluation_model_defined_tags = (
465- create_aqua_evaluation_details .defined_tags or {}
466- )
467469
468470 self .ds_client .update_model (
469471 model_id = evaluation_model .id ,
470472 update_model_details = UpdateModelDetails (
471473 custom_metadata_list = updated_custom_metadata_list ,
472474 freeform_tags = evaluation_model_freeform_tags ,
473- defined_tags = evaluation_model_defined_tags ,
474475 ),
475476 )
476477
0 commit comments