File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 34
34
@patch ("sagemaker_pytorch_serving_container.torchserve._retrieve_ts_server_process" )
35
35
@patch ("sagemaker_pytorch_serving_container.torchserve._add_sigterm_handler" )
36
36
@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 ] )
38
38
@patch ("sagemaker_pytorch_serving_container.torchserve._create_torchserve_config_file" )
39
39
@patch ("sagemaker_pytorch_serving_container.torchserve._adapt_to_ts_format" )
40
40
def test_start_torchserve_default_service_handler (
@@ -77,7 +77,7 @@ def test_start_torchserve_default_service_handler(
77
77
@patch ("sagemaker_pytorch_serving_container.torchserve._retrieve_ts_server_process" )
78
78
@patch ("sagemaker_pytorch_serving_container.torchserve._add_sigterm_handler" )
79
79
@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 ] )
81
81
@patch ("sagemaker_pytorch_serving_container.torchserve._create_torchserve_config_file" )
82
82
@patch ("sagemaker_pytorch_serving_container.torchserve._adapt_to_ts_format" )
83
83
def test_start_torchserve_default_service_handler_multi_model (
You can’t perform that action at this time.
0 commit comments