Skip to content

Commit e8d366f

Browse files
Addressing review comments
1 parent 8edce5c commit e8d366f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ads/aqua/model/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -377,7 +377,7 @@ def delete_model(self, model_id):
377377
f"Failed to delete model:{model_id}. Only registered models or finetuned model can be deleted."
378378
)
379379

380-
@telemetry(entry_point="plugin=model&action=delete", name="aqua")
380+
@telemetry(entry_point="plugin=model&action=edit", name="aqua")
381381
def edit_registered_model(
382382
self, id, inference_container, inference_container_uri, enable_finetuning, task
383383
):
@@ -414,7 +414,7 @@ def edit_registered_model(
414414
if inference_container:
415415
if (
416416
inference_container
417-
in CustomInferenceContainerTypeFamily.values()
417+
in CustomInferenceContainerTypeFamily
418418
and inference_container_uri is None
419419
):
420420
raise AquaRuntimeError(
@@ -431,7 +431,7 @@ def edit_registered_model(
431431
if inference_container_uri:
432432
if (
433433
inference_container
434-
in CustomInferenceContainerTypeFamily.values()
434+
in CustomInferenceContainerTypeFamily
435435
or inference_container is None
436436
):
437437
custom_metadata_list.add(

0 commit comments

Comments
 (0)