updated with formatter changes

Author: iamishaan

ads/model/datascience_model.py

@@ -2176,9 +2176,9 @@ def remove_artifact(
         def find_model_idx():
             for idx, model in enumerate(self.model_file_description["models"]):
                 if (
-                        model["namespace"],
-                        model["bucketName"],
-                        (model["prefix"] if ("prefix" in model) else None),
+                    model["namespace"],
+                    model["bucketName"],
+                    (model["prefix"] if ("prefix" in model) else None),
                 ) == (namespace, bucket, "" if not prefix else prefix):
                     return idx
             return -1

ads/model/service/oci_datascience_model.py

@@ -1,32 +1,35 @@
 #!/usr/bin/env python
+# -*- coding: utf-8; -*-
 
 # Copyright (c) 2022, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import logging
+import time
 from functools import wraps
 from io import BytesIO
 from typing import Callable, Dict, List, Optional
 
 import oci.data_science
+from ads.common import utils
+from ads.common.object_storage_details import ObjectStorageDetails
+from ads.common.oci_datascience import OCIDataScienceMixin
+from ads.common.oci_mixin import OCIWorkRequestMixin
+from ads.common.oci_resource import SEARCH_TYPE, OCIResource
+from ads.common.utils import extract_region
+from ads.common.work_request import DataScienceWorkRequest
+from ads.model.deployment import ModelDeployment
 from oci.data_science.models import (
     ArtifactExportDetailsObjectStorage,
     ArtifactImportDetailsObjectStorage,
     CreateModelDetails,
     ExportModelArtifactDetails,
     ImportModelArtifactDetails,
     UpdateModelDetails,
+    WorkRequest,
 )
 from oci.exceptions import ServiceError
 
-from ads.common.object_storage_details import ObjectStorageDetails
-from ads.common.oci_datascience import OCIDataScienceMixin
-from ads.common.oci_mixin import OCIWorkRequestMixin
-from ads.common.oci_resource import SEARCH_TYPE, OCIResource
-from ads.common.utils import extract_region
-from ads.common.work_request import DataScienceWorkRequest
-from ads.model.deployment import ModelDeployment
-
 logger = logging.getLogger(__name__)
 
 _REQUEST_INTERVAL_IN_SEC = 3