@@ -1051,45 +1051,6 @@ def test_from_model_deployment(
10511051
10521052 assert test_result == test_model
10531053
1054- @patch .object (
1055- ModelDeployment ,
1056- "state" ,
1057- new_callable = PropertyMock ,
1058- return_value = ModelDeploymentState .FAILED ,
1059- )
1060- @patch .object (ModelDeployment , "from_id" )
1061- @patch ("ads.common.auth.default_signer" )
1062- @patch ("ads.common.oci_client.OCIClientFactory" )
1063- def test_from_model_deployment_fail (
1064- self ,
1065- mock_client ,
1066- mock_default_signer ,
1067- mock_from_id ,
1068- mock_model_deployment_state ,
1069- ):
1070- """Tests loading model from model deployment."""
1071- test_auth_config = {"signer" : {"config" : "value" }}
1072- mock_default_signer .return_value = test_auth_config
1073- test_model_deployment_id = "md_ocid"
1074- test_model_id = "model_ocid"
1075- md_props = ModelDeploymentProperties (model_id = test_model_id )
1076- md = ModelDeployment (properties = md_props )
1077- mock_from_id .return_value = md
1078-
1079- with pytest .raises (NotActiveDeploymentError ):
1080- GenericModel .from_model_deployment (
1081- model_deployment_id = test_model_deployment_id ,
1082- model_file_name = "test.pkl" ,
1083- artifact_dir = "test_dir" ,
1084- auth = test_auth_config ,
1085- force_overwrite = True ,
1086- properties = None ,
1087- bucket_uri = "test_bucket_uri" ,
1088- remove_existing_artifact = True ,
1089- compartment_id = "test_compartment_id" ,
1090- )
1091- mock_from_id .assert_called_with (test_model_deployment_id )
1092-
10931054 @patch .object (ModelDeployment , "update" )
10941055 @patch .object (ModelDeployment , "from_id" )
10951056 @patch ("ads.common.auth.default_signer" )
0 commit comments