Remove log

Author: nikhil-sk

src/sagemaker_pytorch_serving_container/handler_service.py

@@ -13,7 +13,6 @@
 from __future__ import absolute_import
 
 from sagemaker_inference.default_handler_service import DefaultHandlerService
-#from sagemaker_inference.transformer import Transformer
 from sagemaker_pytorch_serving_container.transformer import PyTorchTransformer
 from sagemaker_pytorch_serving_container.default_pytorch_inference_handler import DefaultPytorchInferenceHandler
 

src/sagemaker_pytorch_serving_container/torchserve.py

@@ -138,7 +138,7 @@ def _generate_ts_config_properties(handler_service):
     if ts_env.is_env_set() and not ENABLE_MULTI_MODEL:
         models_string = f'''{{\\
         "{DEFAULT_TS_MODEL_NAME}": {{\\
-            "1.0": {{\\
+            "1": {{\\
                 "defaultVersion": true,\\
                 "marName": "{DEFAULT_TS_MODEL_NAME}.mar",\\
                 "minWorkers": {ts_env._min_workers},\\

src/sagemaker_pytorch_serving_container/transformer.py

@@ -28,8 +28,7 @@ def transform(self, data, context):
 
             response_list = []
 
-            for i in range(len(data)):
-                print(f"Processing Data: {data[i]}")
+            for _ in range(len(data)):
                 input_data = data[i].get("body")
 
                 request_processor = context.request_processor[0]