Skip to content

Commit bec0650

Browse files
author
Steven Anton
committed
Fix unit tests
1 parent e4219de commit bec0650

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/unit/test_model_server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
@patch("sagemaker_pytorch_serving_container.torchserve._retrieve_ts_server_process")
3535
@patch("sagemaker_pytorch_serving_container.torchserve._add_sigterm_handler")
3636
@patch("sagemaker_pytorch_serving_container.torchserve._install_requirements")
37-
@patch("os.path.exists", return_value=True)
37+
@patch("os.path.exists", side_effect=[True, False])
3838
@patch("sagemaker_pytorch_serving_container.torchserve._create_torchserve_config_file")
3939
@patch("sagemaker_pytorch_serving_container.torchserve._adapt_to_ts_format")
4040
def test_start_torchserve_default_service_handler(
@@ -77,7 +77,7 @@ def test_start_torchserve_default_service_handler(
7777
@patch("sagemaker_pytorch_serving_container.torchserve._retrieve_ts_server_process")
7878
@patch("sagemaker_pytorch_serving_container.torchserve._add_sigterm_handler")
7979
@patch("sagemaker_pytorch_serving_container.torchserve._install_requirements")
80-
@patch("os.path.exists", return_value=True)
80+
@patch("os.path.exists", side_effect=[True, False])
8181
@patch("sagemaker_pytorch_serving_container.torchserve._create_torchserve_config_file")
8282
@patch("sagemaker_pytorch_serving_container.torchserve._adapt_to_ts_format")
8383
def test_start_torchserve_default_service_handler_multi_model(

0 commit comments

Comments
 (0)