File tree 1 file changed +11
-5
lines changed
1 file changed +11
-5
lines changed Original file line number Diff line number Diff line change @@ -700,16 +700,22 @@ def _mock_serving_variables(
700
700
is_kserve_installed = True ,
701
701
):
702
702
mocker .patch (
703
- "hsml .client.get_serving_resource_limits" ,
703
+ "hopsworks_common .client.get_serving_resource_limits" ,
704
704
return_value = SERVING_RESOURCE_LIMITS ,
705
705
)
706
706
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 ,
708
709
)
709
710
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 ,
711
713
)
712
- mocker .patch ("hsml.client.is_saas_connection" , return_value = is_saas_connection )
713
714
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 ,
715
721
)
You can’t perform that action at this time.
0 commit comments