Ran black formatter of UTs.

Author: YashPandit4u

tests/unitary/default_setup/model/test_datascience_model.py

@@ -164,66 +164,67 @@
 }
 
 MODEL_BY_REF_JSON = {
-  "version": "1.0",
-  "type": "modelOSSReferenceDescription",
-  "models": [
-    {
-      "namespace": "ociodscdev",
-      "bucketName": "unzip-multi-model",
-      "prefix": "model-linear-1",
-      "objects": [
+    "version": "1.0",
+    "type": "modelOSSReferenceDescription",
+    "models": [
         {
-          "name": "model-linear-1/linear-1.pkl",
-          "version": "ee260f4a-920a-4b4e-974a-c13a1032558e",
-          "sizeInBytes": 565
-        }
-      ]
-    },
-    {
-      "namespace": "ociodscdev",
-      "bucketName": "unzip-multi-model",
-      "prefix": "model-linear-2",
-      "objects": [
-        {
-          "name": "model-linear-2/linear-2.pkl",
-          "version": "dc26a7d2-8041-4b37-8ed0-9e8c10869340",
-          "sizeInBytes": 565
-        }
-      ]
-    },
-    {
-      "namespace": "ociodscdev",
-      "bucketName": "unzip-multi-model",
-      "prefix": "model-linear-3",
-      "objects": [
+            "namespace": "ociodscdev",
+            "bucketName": "unzip-multi-model",
+            "prefix": "model-linear-1",
+            "objects": [
+                {
+                    "name": "model-linear-1/linear-1.pkl",
+                    "version": "ee260f4a-920a-4b4e-974a-c13a1032558e",
+                    "sizeInBytes": 565,
+                }
+            ],
+        },
         {
-          "name": "model-linear-3/linear-3.pkl",
-          "version": "a22c1211-f7d4-4fd4-96d8-4e3a048c5cf7",
-          "sizeInBytes": 565
-        }
-      ]
-    },
-    {
-      "namespace": "ociodscdev",
-      "bucketName": "unzip-multi-model",
-      "prefix": "",
-      "objects": [
+            "namespace": "ociodscdev",
+            "bucketName": "unzip-multi-model",
+            "prefix": "model-linear-2",
+            "objects": [
+                {
+                    "name": "model-linear-2/linear-2.pkl",
+                    "version": "dc26a7d2-8041-4b37-8ed0-9e8c10869340",
+                    "sizeInBytes": 565,
+                }
+            ],
+        },
         {
-          "name": "runtime.yaml",
-          "version": "30afb1a6-ab1f-42a3-95e3-09f61a0046fd",
-          "sizeInBytes": 334
+            "namespace": "ociodscdev",
+            "bucketName": "unzip-multi-model",
+            "prefix": "model-linear-3",
+            "objects": [
+                {
+                    "name": "model-linear-3/linear-3.pkl",
+                    "version": "a22c1211-f7d4-4fd4-96d8-4e3a048c5cf7",
+                    "sizeInBytes": 565,
+                }
+            ],
         },
         {
-          "name": "score.py",
-          "version": "c4ccaf96-05be-4174-ac3b-15dce2f558fe",
-          "sizeInBytes": 772
-        }
-      ]
-    }
-  ]
+            "namespace": "ociodscdev",
+            "bucketName": "unzip-multi-model",
+            "prefix": "",
+            "objects": [
+                {
+                    "name": "runtime.yaml",
+                    "version": "30afb1a6-ab1f-42a3-95e3-09f61a0046fd",
+                    "sizeInBytes": 334,
+                },
+                {
+                    "name": "score.py",
+                    "version": "c4ccaf96-05be-4174-ac3b-15dce2f558fe",
+                    "sizeInBytes": 772,
+                },
+            ],
+        },
+    ],
 }
 CONST_MODEL_FILE_DESCRIPTION = "modelDescription"
 
+
 class TestDataScienceModel:
     DEFAULT_PROPERTIES_PAYLOAD = {
         "compartmentId": DSC_MODEL_PAYLOAD["compartmentId"],
@@ -1131,8 +1132,6 @@ def test_download_artifact_for_model_created_by_reference(
 
         mock_large_download.assert_called()
 
-    
-
     @patch("ads.common.oci_client.OCIClientFactory")
     def test_add_artifact(self, mock_oci_client_factory):
         r = ObjectVersionSummary()
@@ -1151,54 +1150,64 @@ def test_add_artifact(self, mock_oci_client_factory):
         # Mock object storage client
         mock_object_storage_client = MagicMock()
         mock_object_storage_client.list_object_versions.return_value = mock_response
-        
+
         mock_oci_client_factory.return_value.object_storage = mock_object_storage_client
 
         # self.mock_dsc_model
         # self.mock_dsc_model.add_artifact(uri="oci://bucket@namespace/prefix")
-        self.mock_dsc_model.add_artifact(namespace="namespace", bucket="bucket", prefix="prefix")
+        self.mock_dsc_model.add_artifact(
+            namespace="namespace", bucket="bucket", prefix="prefix"
+        )
         expected_out = {
-            'version': '1.0', 
-            'type': 'modelOSSReferenceDescription', 
-            'models': [
+            "version": "1.0",
+            "type": "modelOSSReferenceDescription",
+            "models": [
                 {
-                    'namespace': 'namespace', 
-                    'bucketName': 'bucket', 
-                    'prefix': 'prefix', 
-                    'objects': [
+                    "namespace": "namespace",
+                    "bucketName": "bucket",
+                    "prefix": "prefix",
+                    "objects": [
                         {
-                            'name': 'model-linear-2/linear-2.pkl', 
-                            'version': 'dc26a7d2-8041-4b37-8ed0-9e8c10869340', 
-                            'sizeInBytes': 566
+                            "name": "model-linear-2/linear-2.pkl",
+                            "version": "dc26a7d2-8041-4b37-8ed0-9e8c10869340",
+                            "sizeInBytes": 566,
                         }
-                    ]
+                    ],
                 }
-            ]
+            ],
         }
         assert self.mock_dsc_model.model_file_description == expected_out
         # self.mock_dsc_model.remove_artifact(uri="oci://bucket@namespace/prefix")
-        self.mock_dsc_model.remove_artifact(namespace="namespace", bucket="bucket", prefix="prefix")
+        self.mock_dsc_model.remove_artifact(
+            namespace="namespace", bucket="bucket", prefix="prefix"
+        )
         assert self.mock_dsc_model.model_file_description != expected_out
         expected_out = {
-            'version': '1.0', 
-            'type': 'modelOSSReferenceDescription', 
-            'models': []
+            "version": "1.0",
+            "type": "modelOSSReferenceDescription",
+            "models": [],
         }
         assert self.mock_dsc_model.model_file_description == expected_out
 
     def test_remove_artifact(self):
         # self.mock_dsc_model.remove_artifact(uri="oci://unzip-multi-model@ociodscdev/model-linear-1")
-        self.mock_dsc_model.remove_artifact(namespace="ociodscdev", bucket="unzip-multi-model", prefix="model-linear-1")
+        self.mock_dsc_model.remove_artifact(
+            namespace="ociodscdev", bucket="unzip-multi-model", prefix="model-linear-1"
+        )
 
         assert self.mock_dsc_model.model_file_description == None
-        
-        self.mock_dsc_model.set_spec(CONST_MODEL_FILE_DESCRIPTION, deepcopy(MODEL_BY_REF_JSON))
+
+        self.mock_dsc_model.set_spec(
+            CONST_MODEL_FILE_DESCRIPTION, deepcopy(MODEL_BY_REF_JSON)
+        )
         assert self.mock_dsc_model.model_file_description == MODEL_BY_REF_JSON
-        
+
         # self.mock_dsc_model.remove_artifact(uri="oci://unzip-multi-model@ociodscdev/model-linear-1")
-        self.mock_dsc_model.remove_artifact(namespace="ociodscdev", bucket="unzip-multi-model", prefix="model-linear-1")
+        self.mock_dsc_model.remove_artifact(
+            namespace="ociodscdev", bucket="unzip-multi-model", prefix="model-linear-1"
+        )
         assert self.mock_dsc_model.model_file_description != MODEL_BY_REF_JSON
-        
+
         exptected_json = deepcopy(MODEL_BY_REF_JSON)
         exptected_json["models"] = exptected_json["models"][1:]
-        assert self.mock_dsc_model.model_file_description == exptected_json
+        assert self.mock_dsc_model.model_file_description == exptected_json