Removing validation for task

Author: kumar-shivam-ranjan

ads/aqua/config/config.py

@@ -30,10 +30,3 @@ def get_evaluation_service_config(
         .get(container, {})
     )
 
-def get_valid_tasks():
-    return [
-        "text_generation",
-        "code_synthesis",
-        "image_text_to_text",
-        "feature_extraction",
-    ]

ads/aqua/extension/model_handler.py

@@ -13,7 +13,6 @@
     get_hf_model_info,
     list_hf_models,
 )
-from ads.aqua.config.config import get_valid_tasks
 from ads.aqua.extension.base_handler import AquaAPIhandler
 from ads.aqua.extension.errors import Errors
 from ads.aqua.model import AquaModelApp
@@ -165,8 +164,6 @@ def put(self, id):
 
         enable_finetuning = input_data.get("enable_finetuning")
         task = input_data.get("task")
-        if task not in get_valid_tasks():
-            raise HTTPError(400, Errors.INVALID_VALUE_OF_PARAMETER.format("task"))
         self.finish(
             AquaModelApp().edit_registered_model(
                 id, inference_container, enable_finetuning, task

ads/aqua/model/model.py

@@ -33,7 +33,6 @@
     read_file,
     upload_folder,
 )
-from ads.aqua.config.config import get_valid_tasks
 from ads.aqua.constants import (
     AQUA_MODEL_ARTIFACT_CONFIG,
     AQUA_MODEL_ARTIFACT_CONFIG_MODEL_NAME,
@@ -408,13 +407,7 @@ def edit_registered_model(self, id, inference_container, enable_finetuning, task
 
                 custom_metadata_list.remove("modelDescription")
                 if task:
-                    if task in get_valid_tasks():
-                        freeform_tags.update({Tags.TASK: task})
-                    else:
-                        raise AquaValueError(
-                            f"Failed to edit model with the given task parameter. Acceptable values are: {get_valid_tasks()}"
-                        )
-
+                    freeform_tags.update({Tags.TASK: task})
                 updated_custom_metadata_list = [
                     Metadata(**metadata)
                     for metadata in custom_metadata_list.to_dict()["data"]