Fix mocking in test_transformer

logicalclocks · Sep 10, 2024 · b30f8b0 · b30f8b0
1 parent dd88d00
commit b30f8b0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git 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,
         )