Skip to content

Commit b320566

Browse files
committed
Fix mocking in test_predictor
1 parent 3eb1974 commit b320566

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

python/tests/test_predictor.py

+11-5
Original file line numberDiff line numberDiff line change
@@ -700,16 +700,22 @@ def _mock_serving_variables(
700700
is_kserve_installed=True,
701701
):
702702
mocker.patch(
703-
"hsml.client.get_serving_resource_limits",
703+
"hopsworks_common.client.get_serving_resource_limits",
704704
return_value=SERVING_RESOURCE_LIMITS,
705705
)
706706
mocker.patch(
707-
"hsml.client.get_serving_num_instances_limits", return_value=num_instances
707+
"hopsworks_common.client.get_serving_num_instances_limits",
708+
return_value=num_instances,
708709
)
709710
mocker.patch(
710-
"hsml.client.is_scale_to_zero_required", return_value=force_scale_to_zero
711+
"hopsworks_common.client.is_scale_to_zero_required",
712+
return_value=force_scale_to_zero,
711713
)
712-
mocker.patch("hsml.client.is_saas_connection", return_value=is_saas_connection)
713714
mocker.patch(
714-
"hsml.client.is_kserve_installed", return_value=is_kserve_installed
715+
"hopsworks_common.client.is_saas_connection",
716+
return_value=is_saas_connection,
717+
)
718+
mocker.patch(
719+
"hopsworks_common.client.is_kserve_installed",
720+
return_value=is_kserve_installed,
715721
)

0 commit comments

Comments
 (0)