Updating put method for register model

kumar-shivam-ranjan · kumar-shivam-ranjan · commit 27023762da66 · 2024-10-29T19:23:12.000+05:30
diff --git a/ads/aqua/extension/model_handler.py b/ads/aqua/extension/model_handler.py
@@ -170,7 +170,7 @@ def put(self, id):
         task = input_data.get("task")
         if task not in get_valid_tasks():
             raise HTTPError(400, Errors.INVALID_VALUE_OF_PARAMETER.format("task"))
-        return self.finish(
+        self.finish(
             AquaModelApp().edit_registered_model(
                 id, inference_container, enable_finetuning, task
             )
diff --git a/ads/aqua/model/model.py b/ads/aqua/model/model.py
@@ -417,7 +417,7 @@ def edit_registered_model(self, id, inference_container, enable_finetuning, task
                     custom_metadata_list=updated_custom_metadata_list,
                     freeform_tags=freeform_tags,
                 )
-                return AquaApp().update_model(id, update_model_details).data
+                AquaApp().update_model(id, update_model_details)
         else:
             raise AquaRuntimeError(
                 f"Failed to edit model:{id}. Only registered unverified models can be edited."

Original file line number	Diff line number	Diff line change
`@@ -170,7 +170,7 @@ def put(self, id):`
`170`	`170`	`task = input_data.get("task")`
`171`	`171`	`if task not in get_valid_tasks():`
`172`	`172`	`raise HTTPError(400, Errors.INVALID_VALUE_OF_PARAMETER.format("task"))`
`173`		`- return self.finish(`
	`173`	`+ self.finish(`
`174`	`174`	`AquaModelApp().edit_registered_model(`
`175`	`175`	`id, inference_container, enable_finetuning, task`
`176`	`176`	`)`
Original file line number	Diff line number	Diff line change
`@@ -417,7 +417,7 @@ def edit_registered_model(self, id, inference_container, enable_finetuning, task`
`417`	`417`	`custom_metadata_list=updated_custom_metadata_list,`
`418`	`418`	`freeform_tags=freeform_tags,`
`419`	`419`	`)`
`420`		`- return AquaApp().update_model(id, update_model_details).data`
	`420`	`+ AquaApp().update_model(id, update_model_details)`
`421`	`421`	`else:`
`422`	`422`	`raise AquaRuntimeError(`
`423`	`423`	`f"Failed to edit model:{id}. Only registered unverified models can be edited."`