Skip to content

Commit 34d6cd1

Browse files
fix patch
1 parent ec9d3fb commit 34d6cd1

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/unitary/with_extras/aqua/test_deployment_handler.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -82,14 +82,14 @@ def test_get_deployment_config_without_id(self, mock_error):
8282
mock_error.assert_called_once()
8383
assert result["status"] == 400
8484

85-
@patch.object(AquaDeploymentApp, "get")
85+
@patch("ads.aqua.deployment.AquaDeploymentApp.get")
8686
def test_get_deployment(self, mock_get):
8787
"""Test get method to return deployment information."""
8888
self.deployment_handler.request.path = "aqua/deployments"
8989
self.deployment_handler.get(id="mock-model-id")
9090
mock_get.assert_called()
9191

92-
@patch.object(AquaDeploymentApp, "list")
92+
@patch("ads.aqua.deployment.AquaDeploymentApp.list")
9393
def test_list_deployment(self, mock_list):
9494
"""Test get method to return a list of model deployments."""
9595
self.deployment_handler.request.path = "aqua/deployments"
@@ -98,7 +98,7 @@ def test_list_deployment(self, mock_list):
9898
compartment_id=TestDataset.USER_COMPARTMENT_ID, project_id=None
9999
)
100100

101-
@patch.object(AquaDeploymentApp, "create")
101+
@patch("ads.aqua.deployment.AquaDeploymentApp.create")
102102
def test_post(self, mock_create):
103103
"""Test post method to create a model deployment."""
104104
self.deployment_handler.get_json_body = MagicMock(
@@ -131,7 +131,7 @@ def setUp(self, ipython_init_mock) -> None:
131131
self.inference_handler.request = MagicMock()
132132
self.inference_handler.finish = MagicMock()
133133

134-
@patch.object(MDInferenceResponse, "get_model_deployment_response")
134+
@patch("ads.aqua.deployment.MDInferenceResponse.get_model_deployment_response")
135135
def test_post(self, mock_get_model_deployment_response):
136136
"""Test post method to return model deployment response."""
137137
self.inference_handler.get_json_body = MagicMock(

0 commit comments

Comments
 (0)