Addressing review comments

kumar-shivam-ranjan · kumar-shivam-ranjan · commit a4a70b3a37d3 · 2025-01-21T00:42:32.000+05:30
diff --git a/ads/aqua/extension/model_handler.py b/ads/aqua/extension/model_handler.py
@@ -150,7 +150,7 @@ def post(self, *args, **kwargs):  # noqa: ARG002
                 ignore_patterns=ignore_patterns,
                 freeform_tags=freeform_tags,
                 defined_tags=defined_tags,
-                task=task
+                task=task,
             )
         )
 
diff --git a/ads/aqua/model/model.py b/ads/aqua/model/model.py
@@ -376,7 +376,7 @@ def delete_model(self, model_id):
                 f"Failed to delete model:{model_id}. Only registered models or finetuned model can be deleted."
             )
 
-    @telemetry(entry_point="plugin=model&action=delete", name="aqua")
+    @telemetry(entry_point="plugin=model&action=edit", name="aqua")
     def edit_registered_model(self, id, inference_container, enable_finetuning, task):
         """Edits the default config of unverified registered model.
 

Original file line number	Diff line number	Diff line change
`@@ -150,7 +150,7 @@ def post(self, args, *kwargs): # noqa: ARG002`
`150`	`150`	`ignore_patterns=ignore_patterns,`
`151`	`151`	`freeform_tags=freeform_tags,`
`152`	`152`	`defined_tags=defined_tags,`
`153`		`- task=task`
	`153`	`+ task=task,`
`154`	`154`	`)`
`155`	`155`	`)`
`156`	`156`
Original file line number	Diff line number	Diff line change
`@@ -376,7 +376,7 @@ def delete_model(self, model_id):`
`376`	`376`	`f"Failed to delete model:{model_id}. Only registered models or finetuned model can be deleted."`
`377`	`377`	`)`
`378`	`378`
`379`		`- @telemetry(entry_point="plugin=model&action=delete", name="aqua")`
	`379`	`+ @telemetry(entry_point="plugin=model&action=edit", name="aqua")`
`380`	`380`	`def edit_registered_model(self, id, inference_container, enable_finetuning, task):`
`381`	`381`	`"""Edits the default config of unverified registered model.`
`382`	`382`