@@ -77,7 +77,7 @@ def start_torchserve(handler_service=DEFAULT_HANDLER_SERVICE):
77
77
78
78
_set_python_path ()
79
79
80
- _create_torchserve_config_file (handler_service )
80
+ _create_torchserve_config_file ()
81
81
82
82
if os .path .exists (REQUIREMENTS_PATH ):
83
83
_install_requirements ()
@@ -118,20 +118,19 @@ def _set_python_path():
118
118
os .environ [PYTHON_PATH_ENV ] = environment .code_dir
119
119
120
120
121
- def _create_torchserve_config_file (handler_service ):
122
- configuration_properties = _generate_ts_config_properties (handler_service )
121
+ def _create_torchserve_config_file ():
122
+ configuration_properties = _generate_ts_config_properties ()
123
123
124
124
utils .write_file (TS_CONFIG_FILE , configuration_properties )
125
125
126
126
127
- def _generate_ts_config_properties (handler_service ):
127
+ def _generate_ts_config_properties ():
128
128
env = environment .Environment ()
129
129
user_defined_configuration = {
130
130
"default_response_timeout" : env .model_server_timeout ,
131
131
"default_workers_per_model" : env .model_server_workers ,
132
132
"inference_address" : "http://0.0.0.0:{}" .format (env .inference_http_port ),
133
133
"management_address" : "http://0.0.0.0:{}" .format (env .management_http_port ),
134
- "default_service_handler" : handler_service ,
135
134
}
136
135
137
136
ts_env = ts_environment .TorchServeEnvironment ()
0 commit comments