ensure model store path exists

Author: mseth10

src/sagemaker_pytorch_serving_container/torchserve.py

@@ -68,8 +68,12 @@ def start_torchserve(handler_service=DEFAULT_HANDLER_SERVICE):
 
     """
 
-    if ENABLE_MULTI_MODEL and "SAGEMAKER_HANDLER" not in os.environ:
-        os.environ["SAGEMAKER_HANDLER"] = handler_service
+    if ENABLE_MULTI_MODEL:
+        if "SAGEMAKER_HANDLER" not in os.environ:
+            os.environ["SAGEMAKER_HANDLER"] = handler_service
+    else:
+        if not os.path.exists(MODEL_STORE):
+            os.makedirs(MODEL_STORE)
 
     _set_python_path()