From b30f8b08c010e8248fc051259bd8b832fc9228c0 Mon Sep 17 00:00:00 2001 From: Aleksey Veresov Date: Tue, 10 Sep 2024 09:43:02 +0200 Subject: [PATCH] Fix mocking in test_transformer --- python/tests/test_transformer.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/python/tests/test_transformer.py b/python/tests/test_transformer.py index 7df302bd6..10150a7f1 100644 --- a/python/tests/test_transformer.py +++ b/python/tests/test_transformer.py @@ -298,12 +298,14 @@ def test_extract_fields_from_json(self, mocker, backend_fixtures): def _mock_serving_variables(self, mocker, num_instances, force_scale_to_zero=False): mocker.patch( - "hsml.client.get_serving_resource_limits", + "hopsworks_common.client.get_serving_resource_limits", return_value=SERVING_RESOURCE_LIMITS, ) mocker.patch( - "hsml.client.get_serving_num_instances_limits", return_value=num_instances + "hopsworks_common.client.get_serving_num_instances_limits", + return_value=num_instances, ) mocker.patch( - "hsml.client.is_scale_to_zero_required", return_value=force_scale_to_zero + "hopsworks_common.client.is_scale_to_zero_required", + return_value=force_scale_to_zero, )