Updated pr.

Author: lu-ohai

ads/model/deployment/model_deployment.py

@@ -70,7 +70,6 @@
 MODEL_DEPLOYMENT_INSTANCE_COUNT = 1
 MODEL_DEPLOYMENT_BANDWIDTH_MBPS = 10
 
-MAX_ARTIFACT_SIZE_IN_BYTES = 2147483648  # 2GB
 
 class ModelDeploymentLogType:
     PREDICT = "predict"
@@ -1567,13 +1566,6 @@ def _build_model_deployment_configuration_details(self) -> Dict:
 
         model_id = runtime.model_uri
         if not model_id.startswith("ocid"):
-            if ads_utils.folder_size(runtime.model_uri) > MAX_ARTIFACT_SIZE_IN_BYTES:
-                if not runtime.bucket_uri:
-                    raise ValueError(
-                        f"The model artifacts size is greater than `{MAX_ARTIFACT_SIZE_IN_BYTES}`. "
-                        "The `bucket_uri` needs to be specified to copy artifacts to the object storage bucket. "
-                        "Example: `runtime.with_bucket_uri(oci://<bucket_name>@<namespace>/prefix/)`"
-                    )
 
             from ads.model.datascience_model import DataScienceModel
 

tests/unitary/default_setup/model_deployment/test_model_deployment_v2.py

@@ -11,7 +11,6 @@
 import unittest
 import pandas
 from unittest.mock import MagicMock, patch
-from ads.common import utils
 from ads.common.oci_datascience import OCIDataScienceMixin
 from ads.common.oci_logging import ConsolidatedLog, OCILog
 from ads.common.oci_mixin import OCIModelMixin
@@ -23,7 +22,6 @@
     ModelDeployment,
     ModelDeploymentLogType,
     ModelDeploymentFailedError,
-    MAX_ARTIFACT_SIZE_IN_BYTES,
 )
 from ads.model.deployment.model_deployment_infrastructure import (
     ModelDeploymentInfrastructure,
@@ -1390,10 +1388,8 @@ def test_model_deployment_with_subnet_id(self):
         "create_model_deployment",
     )
     @patch.object(DataScienceModel, "create")
-    @patch.object(utils, "folder_size")
     def test_model_deployment_with_large_size_artifact(
         self, 
-        mock_folder_size,
         mock_create, 
         mock_create_model_deployment, 
         mock_sync
@@ -1409,6 +1405,7 @@ def test_model_deployment_with_large_size_artifact(
             .with_overwrite_existing_artifact(True)
             .with_remove_existing_artifact(True)
             .with_timeout(100)
+            .with_bucket_uri("test_bucket_uri")
         )
 
         runtime_dict = model_deployment.runtime.to_dict()["spec"]
@@ -1417,20 +1414,14 @@ def test_model_deployment_with_large_size_artifact(
         assert runtime_dict["overwriteExistingArtifact"] == True
         assert runtime_dict["removeExistingArtifact"] == True
         assert runtime_dict["timeout"] == 100
+        assert runtime_dict["bucketUri"] == "test_bucket_uri"
 
         response = MagicMock()
         response.data = OCI_MODEL_DEPLOYMENT_RESPONSE
         mock_create_model_deployment.return_value = response
         model_deployment = self.initialize_model_deployment()
         model_deployment.set_spec(model_deployment.CONST_ID, "test_model_deployment_id")
 
-        mock_folder_size.return_value = MAX_ARTIFACT_SIZE_IN_BYTES + 1
-        with pytest.raises(ValueError):
-            model_deployment.deploy(wait_for_completion=False)
-
-        model_deployment.runtime.with_bucket_uri("test_bucket_uri")
-        runtime_dict = model_deployment.runtime.to_dict()["spec"]
-        assert runtime_dict["bucketUri"] == "test_bucket_uri"
         create_model_deployment_details = (
             model_deployment._build_model_deployment_details()
         )