ODSC-50486/print out model deployment id when calling predict (#491)

Author: mingkang111

ads/model/service/oci_datascience_model_deployment.py

@@ -67,11 +67,11 @@ def wrapper(self, *args, **kwargs):
     return decorator
 
 
-class MissingModelDeploymentIdError(Exception):   # pragma: no cover
+class MissingModelDeploymentIdError(Exception):  # pragma: no cover
     pass
 
 
-class MissingModelDeploymentWorkflowIdError(Exception):   # pragma: no cover
+class MissingModelDeploymentWorkflowIdError(Exception):  # pragma: no cover
     pass
 
 
@@ -175,32 +175,23 @@ def activate(
             The `OCIDataScienceModelDeployment` instance (self).
         """
         dsc_model_deployment = OCIDataScienceModelDeployment.from_id(self.id)
-        if (
-            dsc_model_deployment.lifecycle_state
-            == self.LIFECYCLE_STATE_ACTIVE
-        ):
+        if dsc_model_deployment.lifecycle_state == self.LIFECYCLE_STATE_ACTIVE:
             raise Exception(
                 f"Model deployment {dsc_model_deployment.id} is already in active state."
             )
 
-        if (
-            dsc_model_deployment.lifecycle_state
-            == self.LIFECYCLE_STATE_INACTIVE
-        ):
+        if dsc_model_deployment.lifecycle_state == self.LIFECYCLE_STATE_INACTIVE:
             logger.info(f"Activating model deployment `{self.id}`.")
             response = self.client.activate_model_deployment(
                 self.id,
             )
 
             if wait_for_completion:
-
                 self.workflow_req_id = response.headers.get("opc-work-request-id", None)
 
                 try:
                     self.wait_for_progress(
-                        self.workflow_req_id, 
-                        max_wait_time, 
-                        poll_interval
+                        self.workflow_req_id, max_wait_time, poll_interval
                     )
                 except Exception as e:
                     logger.error(
@@ -241,21 +232,17 @@ def create(
         response = self.client.create_model_deployment(create_model_deployment_details)
         self.update_from_oci_model(response.data)
         logger.info(f"Creating model deployment `{self.id}`.")
+        print(f"Model Deployment OCID: {self.id}")
 
         if wait_for_completion:
-
             self.workflow_req_id = response.headers.get("opc-work-request-id", None)
 
             try:
                 self.wait_for_progress(
-                    self.workflow_req_id, 
-                    max_wait_time, 
-                    poll_interval
+                    self.workflow_req_id, max_wait_time, poll_interval
                 )
             except Exception as e:
-                logger.error(
-                    "Error while trying to create model deployment: " + str(e)
-                )
+                logger.error("Error while trying to create model deployment: " + str(e))
 
         return self.sync()
 
@@ -287,32 +274,23 @@ def deactivate(
             The `OCIDataScienceModelDeployment` instance (self).
         """
         dsc_model_deployment = OCIDataScienceModelDeployment.from_id(self.id)
-        if (
-            dsc_model_deployment.lifecycle_state
-            == self.LIFECYCLE_STATE_INACTIVE
-        ):
+        if dsc_model_deployment.lifecycle_state == self.LIFECYCLE_STATE_INACTIVE:
             raise Exception(
                 f"Model deployment {dsc_model_deployment.id} is already in inactive state."
             )
 
-        if (
-            dsc_model_deployment.lifecycle_state
-            == self.LIFECYCLE_STATE_ACTIVE
-        ):
+        if dsc_model_deployment.lifecycle_state == self.LIFECYCLE_STATE_ACTIVE:
             logger.info(f"Deactivating model deployment `{self.id}`.")
             response = self.client.deactivate_model_deployment(
                 self.id,
             )
 
             if wait_for_completion:
-
                 self.workflow_req_id = response.headers.get("opc-work-request-id", None)
 
                 try:
                     self.wait_for_progress(
-                        self.workflow_req_id, 
-                        max_wait_time, 
-                        poll_interval
+                        self.workflow_req_id, max_wait_time, poll_interval
                     )
                 except Exception as e:
                     logger.error(
@@ -355,7 +333,7 @@ def delete(
         dsc_model_deployment = OCIDataScienceModelDeployment.from_id(self.id)
         if dsc_model_deployment.lifecycle_state in [
             self.LIFECYCLE_STATE_DELETED,
-            self.LIFECYCLE_STATE_DELETING
+            self.LIFECYCLE_STATE_DELETING,
         ]:
             raise Exception(
                 f"Model deployment {dsc_model_deployment.id} is either deleted or being deleted."
@@ -374,19 +352,14 @@ def delete(
         )
 
         if wait_for_completion:
-
             self.workflow_req_id = response.headers.get("opc-work-request-id", None)
 
             try:
                 self.wait_for_progress(
-                    self.workflow_req_id, 
-                    max_wait_time, 
-                    poll_interval
+                    self.workflow_req_id, max_wait_time, poll_interval
                 )
             except Exception as e:
-                logger.error(
-                    "Error while trying to delete model deployment: " + str(e)
-                )
+                logger.error("Error while trying to delete model deployment: " + str(e))
 
         return self.sync()
 
@@ -440,9 +413,7 @@ def update(
             )
             self.workflow_req_id = response.headers.get("opc-work-request-id", None)
         except Exception as e:
-            logger.error(
-                "Error while trying to update model deployment: " + str(e)
-            )
+            logger.error("Error while trying to update model deployment: " + str(e))
 
         return self.sync()