Formatting

Author: kumar-shivam-ranjan

ads/aqua/common/enums.py

@@ -53,8 +53,10 @@ class InferenceContainerTypeFamily(str, metaclass=ExtendedEnumMeta):
     AQUA_TGI_CONTAINER_FAMILY = "odsc-tgi-serving"
     AQUA_LLAMA_CPP_CONTAINER_FAMILY = "odsc-llama-cpp-serving"
 
-class CustomInferenceContainerTypeFamily(str,metaclass=ExtendedEnumMeta):
-    AQUA_TEI_CONTAINER_FAMILY="odsc-tei-serving"
+
+class CustomInferenceContainerTypeFamily(str, metaclass=ExtendedEnumMeta):
+    AQUA_TEI_CONTAINER_FAMILY = "odsc-tei-serving"
+
 
 class InferenceContainerParamType(str, metaclass=ExtendedEnumMeta):
     PARAM_TYPE_VLLM = "VLLM_PARAMS"

ads/aqua/extension/model_handler.py

@@ -8,7 +8,9 @@
 from tornado.web import HTTPError
 
 from ads.aqua.common.decorator import handle_exceptions
-from ads.aqua.common.enums import InferenceContainerTypeFamily,CustomInferenceContainerTypeFamily
+from ads.aqua.common.enums import (
+    CustomInferenceContainerTypeFamily,
+)
 from ads.aqua.common.errors import AquaRuntimeError, AquaValueError
 from ads.aqua.common.utils import (
     get_hf_model_info,

ads/aqua/model/model.py

@@ -15,9 +15,10 @@
 from ads.aqua import ODSC_MODEL_COMPARTMENT_OCID, logger
 from ads.aqua.app import AquaApp
 from ads.aqua.common.enums import (
+    CustomInferenceContainerTypeFamily,
     FineTuningContainerTypeFamily,
     InferenceContainerTypeFamily,
-    Tags, CustomInferenceContainerTypeFamily,
+    Tags,
 )
 from ads.aqua.common.errors import AquaRuntimeError, AquaValueError
 from ads.aqua.common.utils import (
@@ -405,18 +406,19 @@ def edit_registered_model(
         if ds_model.freeform_tags.get(Tags.BASE_MODEL_CUSTOM, None):
             if ds_model.freeform_tags.get(Tags.AQUA_SERVICE_MODEL_TAG, None):
                 raise AquaRuntimeError(
-                    f"Only registered unverified models can be edited."
+                    "Only registered unverified models can be edited."
                 )
             else:
                 custom_metadata_list = ds_model.custom_metadata_list
                 freeform_tags = ds_model.freeform_tags
                 if inference_container:
                     if (
-                            inference_container in CustomInferenceContainerTypeFamily.values()
+                        inference_container
+                        in CustomInferenceContainerTypeFamily.values()
                         and inference_container_uri is None
                     ):
                         raise AquaRuntimeError(
-                            f"Inference container URI must be provided."
+                            "Inference container URI must be provided."
                         )
                     else:
                         custom_metadata_list.add(
@@ -428,7 +430,8 @@ def edit_registered_model(
                         )
                 if inference_container_uri:
                     if (
-                            inference_container in CustomInferenceContainerTypeFamily.values()
+                        inference_container
+                        in CustomInferenceContainerTypeFamily.values()
                         or inference_container is None
                     ):
                         custom_metadata_list.add(
@@ -477,9 +480,7 @@ def edit_registered_model(
                 )
                 AquaApp().update_model(id, update_model_details)
         else:
-            raise AquaRuntimeError(
-                f"Only registered unverified models can be edited."
-            )
+            raise AquaRuntimeError("Only registered unverified models can be edited.")
 
     def _fetch_metric_from_metadata(
         self,