@@ -52,6 +52,7 @@ def __init__(self):
52
52
53
53
@patch ('sagemaker_inference.environment.Environment' )
54
54
@patch ('importlib.util.module_from_spec' , return_value = UserModuleTransformFn ())
55
+ @patch ('os.path.exists' , return_value = True )
55
56
def test_user_module_transform_fn (import_module , env ):
56
57
env .return_value .module_name = MODULE_NAME
57
58
transformer = HandlerService ._user_module_transformer ()
@@ -68,6 +69,7 @@ def __init__(self):
68
69
69
70
@patch ('sagemaker_inference.environment.Environment' )
70
71
@patch ('importlib.util.module_from_spec' , return_value = UserModuleModelFn ())
72
+ @patch ('os.path.exists' , return_value = True )
71
73
def test_user_module_mxnet_module_transformer (import_module , env ):
72
74
env .return_value .module_name = MODULE_NAME
73
75
import_module .return_value .model_fn .return_value = mx .module .BaseModule ()
@@ -81,6 +83,7 @@ def test_user_module_mxnet_module_transformer(import_module, env):
81
83
@patch ('sagemaker_inference.environment.Environment' )
82
84
@patch ('sagemaker_mxnet_serving_container.default_inference_handler.DefaultMXNetInferenceHandler.default_model_fn' )
83
85
@patch ('importlib.util.module_from_spec' , return_value = object ())
86
+ @patch ('os.path.exists' , return_value = True )
84
87
def test_default_inference_handler_mxnet_gluon_transformer (import_module , model_fn , env ):
85
88
env .return_value .module_name = MODULE_NAME
86
89
model_fn .return_value = mx .gluon .block .Block ()
0 commit comments