Skip to content

Commit 4729dcc

Browse files
committed
Fix pattern
1 parent 26b9d86 commit 4729dcc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/unit/test_default_inference_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def test_default_model_fn_non_torchscript_model(inference_handler, test_case):
112112
mock_os.path.exists.return_value = True
113113
with mock.patch("torch.jit") as mock_torch_jit:
114114
mock_torch_jit.load = _produce_runtime_error
115-
with pytest.raises(Exception, match=r"Failed to load \s*. Please ensure model is saved using torchscript."):
115+
with pytest.raises(Exception, match=r"Failed to load .*. Please ensure model is saved using torchscript."):
116116
inference_handler.default_model_fn("model_dir")
117117

118118

0 commit comments

Comments
 (0)