Fixed unit tests.

Author: lu-ohai

ads/model/deployment/model_deployment.py

@@ -920,11 +920,10 @@ def predict(
                 "`data` and `json_input` are both provided. You can only use one of them."
             )
 
-        self._validate_bandwidth(data or json_input)
-
         if auto_serialize_data:
             data = data or json_input
             serialized_data = serializer.serialize(data=data)
+            self._validate_bandwidth(serialized_data)
             return send_request(
                 data=serialized_data,
                 endpoint=endpoint,
@@ -957,6 +956,7 @@ def predict(
             raise ValueError(
                 "`model_name` and `model_version` have to be provided together."
             )
+        self._validate_bandwidth(data)
         prediction = send_request(
             data=data,
             endpoint=endpoint,

tests/unitary/default_setup/model_deployment/test_model_deployment.py

@@ -14,13 +14,17 @@
     ModelDeployment,
     ModelDeploymentProperties,
 )
+from ads.model.deployment.model_deployment_infrastructure import ModelDeploymentInfrastructure
+from ads.model.deployment.model_deployment_runtime import ModelDeploymentCondaRuntime
 
 
 class ModelDeploymentTestCase(unittest.TestCase):
     MODEL_ID = "<MODEL_OCID>"
     with patch.object(oci_client, "OCIClientFactory"):
         test_model_deployment = ModelDeployment(
-            model_deployment_id="test_model_deployment_id", properties={}
+            model_deployment_id="test_model_deployment_id", properties={},
+            infrastructure=ModelDeploymentInfrastructure(),
+            runtime=ModelDeploymentCondaRuntime()
         )
 
     @patch("requests.post")