Skip to content

Commit d62f5ff

Browse files
authored
revert passing handler service to ts config
1 parent 1291d87 commit d62f5ff

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

src/sagemaker_pytorch_serving_container/torchserve.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def start_torchserve(handler_service=DEFAULT_HANDLER_SERVICE):
7777

7878
_set_python_path()
7979

80-
_create_torchserve_config_file(handler_service)
80+
_create_torchserve_config_file()
8181

8282
if os.path.exists(REQUIREMENTS_PATH):
8383
_install_requirements()
@@ -118,20 +118,19 @@ def _set_python_path():
118118
os.environ[PYTHON_PATH_ENV] = environment.code_dir
119119

120120

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()
123123

124124
utils.write_file(TS_CONFIG_FILE, configuration_properties)
125125

126126

127-
def _generate_ts_config_properties(handler_service):
127+
def _generate_ts_config_properties():
128128
env = environment.Environment()
129129
user_defined_configuration = {
130130
"default_response_timeout": env.model_server_timeout,
131131
"default_workers_per_model": env.model_server_workers,
132132
"inference_address": "http://0.0.0.0:{}".format(env.inference_http_port),
133133
"management_address": "http://0.0.0.0:{}".format(env.management_http_port),
134-
"default_service_handler": handler_service,
135134
}
136135

137136
ts_env = ts_environment.TorchServeEnvironment()

0 commit comments

Comments
 (0)