Formatting

Author: kumar-shivam-ranjan

ads/aqua/extension/model_handler.py

@@ -13,6 +13,7 @@
     get_hf_model_info,
     list_hf_models,
 )
+from ads.aqua.common.enums import InferenceContainerTypeFamily
 from ads.aqua.extension.base_handler import AquaAPIhandler
 from ads.aqua.extension.errors import Errors
 from ads.aqua.model import AquaModelApp
@@ -163,12 +164,13 @@ def put(self, id):
             raise HTTPError(400, Errors.NO_INPUT_DATA)
 
         inference_container = input_data.get("inference_container")
-        inference_container_uri=input_data.get("inference_container_uri")
+        inference_container_uri = input_data.get("inference_container_uri")
         inference_containers = AquaModelApp.list_valid_inference_containers()
         if (
             inference_container is not None
             and inference_container not in inference_containers
-            and inference_container != InferenceContainerTypeFamily.AQUA_TEI_CONTAINER_FAMILY
+            and inference_container
+            != InferenceContainerTypeFamily.AQUA_TEI_CONTAINER_FAMILY
         ):
             raise HTTPError(
                 400, Errors.INVALID_VALUE_OF_PARAMETER.format("inference_container")
@@ -178,7 +180,13 @@ def put(self, id):
         task = input_data.get("task")
         app = AquaModelApp()
         self.finish(
-            app.edit_registered_model(id, inference_container, inference_container_uri,enable_finetuning, task)
+            app.edit_registered_model(
+                id,
+                inference_container,
+                inference_container_uri,
+                enable_finetuning,
+                task,
+            )
         )
         app.clear_model_details_cache(model_id=id)
 

ads/aqua/model/model.py

@@ -377,7 +377,9 @@ def delete_model(self, model_id):
             )
 
     @telemetry(entry_point="plugin=model&action=delete", name="aqua")
-    def edit_registered_model(self, id, inference_container,inference_container_uri, enable_finetuning, task):
+    def edit_registered_model(
+        self, id, inference_container, inference_container_uri, enable_finetuning, task
+    ):
         """Edits the default config of unverified registered model.
 
         Parameters
@@ -421,7 +423,7 @@ def edit_registered_model(self, id, inference_container,inference_container_uri,
                         key=ModelCustomMetadataFields.DEPLOYMENT_CONTAINER_URI,
                         value=inference_container_uri,
                         category=MetadataCustomCategory.OTHER,
-                        description=f"Inference container URI for {ds_model.display_name}"
+                        description=f"Inference container URI for {ds_model.display_name}",
                     )
                 if enable_finetuning is not None:
                     if enable_finetuning.lower() == "true":