Skip to content

Commit 2702376

Browse files
Updating put method for register model
1 parent e0f0f3b commit 2702376

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

ads/aqua/extension/model_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def put(self, id):
170170
task = input_data.get("task")
171171
if task not in get_valid_tasks():
172172
raise HTTPError(400, Errors.INVALID_VALUE_OF_PARAMETER.format("task"))
173-
return self.finish(
173+
self.finish(
174174
AquaModelApp().edit_registered_model(
175175
id, inference_container, enable_finetuning, task
176176
)

ads/aqua/model/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -417,7 +417,7 @@ def edit_registered_model(self, id, inference_container, enable_finetuning, task
417417
custom_metadata_list=updated_custom_metadata_list,
418418
freeform_tags=freeform_tags,
419419
)
420-
return AquaApp().update_model(id, update_model_details).data
420+
AquaApp().update_model(id, update_model_details)
421421
else:
422422
raise AquaRuntimeError(
423423
f"Failed to edit model:{id}. Only registered unverified models can be edited."

0 commit comments

Comments
 (0)