Skip to content

Commit cad3d46

Browse files
committed
Updated pr.
1 parent 6008077 commit cad3d46

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/unitary/default_setup/model/test_oci_datascience_model.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424
from ads.common.oci_resource import SEARCH_TYPE, OCIResource
2525
from ads.dataset.progress import TqdmProgressBar
2626
from ads.model.datascience_model import _MAX_ARTIFACT_SIZE_IN_BYTES
27-
from ads.model.deployment.model_deployer import ModelDeployer
2827
from ads.model.service.oci_datascience_model import (
2928
ModelArtifactNotFoundError,
3029
ModelProvenanceNotFoundError,
@@ -230,10 +229,10 @@ def test_delete_success(self, mock_client):
230229
mock_model_deployment.return_value = [
231230
MagicMock(lifecycle_state="ACTIVE", identifier="md_id")
232231
]
233-
with patch.object(ModelDeployer, "delete") as mock_md_delete:
232+
with patch("ads.model.deployment.ModelDeployment.from_id") as mock_from_id:
234233
with patch.object(OCIDataScienceModel, "sync") as mock_sync:
235234
self.mock_model.delete(delete_associated_model_deployment=True)
236-
mock_md_delete.assert_called_with(model_deployment_id="md_id")
235+
mock_from_id.assert_called_with("md_id")
237236
mock_client.delete_model.assert_called_with(self.mock_model.id)
238237
mock_sync.assert_called()
239238

0 commit comments

Comments
 (0)