From cc58bef268fe755deb6699e67674a14929d7bc19 Mon Sep 17 00:00:00 2001 From: Michael Silver <mikesterlingw@users.noreply.github.com> Date: Mon, 21 Feb 2022 15:44:43 +0100 Subject: [PATCH] Reference PyTorch instead of MXNet Documentation was erroneously referring to MXNet despite this handler service being written for PyTorch --- src/sagemaker_pytorch_serving_container/handler_service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker_pytorch_serving_container/handler_service.py b/src/sagemaker_pytorch_serving_container/handler_service.py index 07e81c7b..1d9e16c6 100644 --- a/src/sagemaker_pytorch_serving_container/handler_service.py +++ b/src/sagemaker_pytorch_serving_container/handler_service.py @@ -26,13 +26,13 @@ class HandlerService(DefaultHandlerService): """Handler service that is executed by the model server. - Determines specific default inference handlers to use based on the type MXNet model being used. + Determines specific default inference handlers to use based on the type of PyTorch model being used. This class extends ``DefaultHandlerService``, which define the following: - The ``handle`` method is invoked for all incoming inference requests to the model server. - The ``initialize`` method is invoked at model server start up. - Based on: https://github.com/awslabs/mxnet-model-server/blob/master/docs/custom_service.md + Based on: https://github.com/awslabs/multi-model-server/blob/master/docs/custom_service.md """ def __init__(self):