Skip to content

Commit 2be5474

Browse files
authored
Update test_model_server.py
1 parent 9262b48 commit 2be5474

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/unit/test_model_server.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -275,10 +275,11 @@ def test_add_sigterm_handler(signal_call):
275275
@patch("subprocess.check_call")
276276
def test_install_requirements(check_call):
277277
torchserve._install_requirements()
278-
for i in ['pip', 'install', '-r', '/opt/ml/model/code/requirements.txt']:
278+
for i in ['pip', 'install', '-r', '/opt/ml/model/code/requirements.txt']:
279279
assert i in check_call.call_args.args[0]
280280

281-
@patch("subprocess.check_call", side_effect=subprocess.CalledProcessError(0, "cmd"))
281+
282+
@patch("subprocess.check_call", side_effect=subprocess.CalledProcessError(0, "cmd"))
282283
def test_install_requirements_installation_failed(check_call):
283284
with pytest.raises(ValueError) as e:
284285
torchserve._install_requirements()

0 commit comments

Comments
 (0)