File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 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" )
4040def 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" )
8383def test_start_torchserve_default_service_handler_multi_model (
You can’t perform that action at this time.
0 commit comments