Skip to content

Commit 46eb97d

Browse files
committed
Add defined_tag when creating the model instead of updating the model.
1 parent 27edc84 commit 46eb97d

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

ads/aqua/evaluation/evaluation.py

+11-10
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)