formatting

Author: kumar-shivam-ranjan

ads/aqua/model/model.py

@@ -419,7 +419,7 @@ def edit_registered_model(self, id, inference_container, enable_finetuning, task
                 return AquaApp().update_model(id, update_model_details).data
         else:
             raise AquaRuntimeError(
-                f"Failed to edit model:{id}. Only registered unverified models can be deleted."
+                f"Failed to edit model:{id}. Only registered unverified models can be edited."
             )
 
     def _fetch_metric_from_metadata(

tests/unitary/with_extras/aqua/test_deployment_handler.py

@@ -93,22 +93,26 @@ def test_get_deployment(self, mock_get):
         mock_get.assert_called()
 
     @patch("ads.aqua.modeldeployment.AquaDeploymentApp.delete")
-    def test_delete_deployment(self,mock_delete):
+    def test_delete_deployment(self, mock_delete):
         self.deployment_handler.request.path = "aqua/deployments"
         self.deployment_handler.delete("mock-model-id")
         mock_delete.assert_called()
 
     @patch("ads.aqua.modeldeployment.AquaDeploymentApp.activate")
-    def test_activate_deployment(self,mock_activate):
-        self.deployment_handler.request.path = "aqua/deployments/ocid1.datasciencemodeldeployment.oc1.iad.xxx/activate"
-        mock_activate.return_value={"lifecycle_state":"UPDATING"}
+    def test_activate_deployment(self, mock_activate):
+        self.deployment_handler.request.path = (
+            "aqua/deployments/ocid1.datasciencemodeldeployment.oc1.iad.xxx/activate"
+        )
+        mock_activate.return_value = {"lifecycle_state": "UPDATING"}
         self.deployment_handler.put()
         mock_activate.assert_called()
 
     @patch("ads.aqua.modeldeployment.AquaDeploymentApp.deactivate")
-    def test_deactivate_deployment(self,mock_deactivate):
-        self.deployment_handler.request.path = "aqua/deployments/ocid1.datasciencemodeldeployment.oc1.iad.xxx/deactivate"
-        mock_deactivate.return_value={"lifecycle_state":"UPDATING"}
+    def test_deactivate_deployment(self, mock_deactivate):
+        self.deployment_handler.request.path = (
+            "aqua/deployments/ocid1.datasciencemodeldeployment.oc1.iad.xxx/deactivate"
+        )
+        mock_deactivate.return_value = {"lifecycle_state": "UPDATING"}
         self.deployment_handler.put()
         mock_deactivate.assert_called()
 

tests/unitary/with_extras/aqua/test_model_handler.py

@@ -94,10 +94,10 @@ def test_delete_with_id(self, mock_delete, mock_urlparse):
             mock_urlparse.assert_called()
             mock_delete.assert_called()
 
-    @patch.object(AquaContainerConfig,"from_container_index_json")
-    @patch.object(AquaModelApp,"edit_registered_model")
-    def test_put(self,mock_edit,mock_container_index):
-        mock_edit.return_value={"state":"EDITED"}
+    @patch.object(AquaContainerConfig, "from_container_index_json")
+    @patch.object(AquaModelApp, "edit_registered_model")
+    def test_put(self, mock_edit, mock_container_index):
+        mock_edit.return_value = {"state": "EDITED"}
         mock_inference = MagicMock()
         mock_inference.values.return_value = [
             MagicMock(family="odsc-vllm-serving"),
@@ -113,10 +113,12 @@ def test_put(self,mock_edit,mock_container_index):
                 inference_container="odsc-tgi-serving",
             )
         )
-        with patch("ads.aqua.extension.base_handler.AquaAPIhandler.finish") as mock_finish:
+        with patch(
+            "ads.aqua.extension.base_handler.AquaAPIhandler.finish"
+        ) as mock_finish:
             mock_finish.side_effect = lambda x: x
             result = self.model_handler.put(id="ocid1.datasciencemodel.oc1.iad.xxx")
-            print(f"result: ",result)
+            print(f"result: ", result)
             assert result["state"] is "EDITED"
             mock_edit.assert_called()