Skip to content

Commit fb60ed0

Browse files
Adding review comments
1 parent 5688652 commit fb60ed0

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

ads/aqua/extension/deployment_handler.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -291,5 +291,7 @@ def post(self, *args, **kwargs):
291291
("deployments/?([^/]*)/params", AquaDeploymentParamsHandler),
292292
("deployments/config/?([^/]*)", AquaDeploymentHandler),
293293
("deployments/?([^/]*)", AquaDeploymentHandler),
294+
("deployments/?([^/]*)/activate", AquaDeploymentHandler),
295+
("deployments/?([^/]*)/deactivate", AquaDeploymentHandler),
294296
("inference", AquaDeploymentInferenceHandler),
295297
]

ads/aqua/model/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -404,7 +404,7 @@ def edit_registered_model(self, id, inference_container, enable_finetuning, task
404404

405405
custom_metadata_list.remove("modelDescription")
406406
if task:
407-
freeform_tags.update({"task": task})
407+
freeform_tags.update({Tags.TASK: task})
408408

409409
updated_custom_metadata_list = [
410410
Metadata(**metadata)

0 commit comments

Comments
 (0)