Skip to content

Commit 35af9cd

Browse files
committed
Fixed unit test.
1 parent 5cfdb4f commit 35af9cd

File tree

1 file changed

+0
-39
lines changed

1 file changed

+0
-39
lines changed

tests/unitary/with_extras/model/test_generic_model.py

Lines changed: 0 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)