fixing modules

Author: Bryannah Hernandez

src/sagemaker/pipeline.py

@@ -43,7 +43,6 @@
 from sagemaker.utils import instance_supports_kms
 
 
-
 class PipelineModel(object):
     """A pipeline of SageMaker `Model` instances.
 

src/sagemaker/serve/app.py

@@ -3,6 +3,18 @@
 from __future__ import absolute_import
 
 import logging
+import importlib.util
+
+if not importlib.util.find_spec("uvicorn"):
+    raise ImportError("Unable to import uvicorn, check if uvicorn is installed")
+
+if not importlib.util.find_spec("transformers"):
+    raise ImportError("Unable to import transformers, check if transformers is installed")
+
+if not importlib.util.find_spec("fastapi"):
+    raise ImportError("Unable to import fastapi, check if fastapi is installed")
+
+import uvicorn
 from transformers import pipeline
 from fastapi import FastAPI, Request
 

src/sagemaker/serve/model_server/multi_model_server/server.py

@@ -16,7 +16,6 @@
 from sagemaker.s3 import S3Uploader
 from sagemaker.local.utils import get_docker_host
 from sagemaker.serve.utils.optimize_utils import _is_s3_uri
-from sagemaker.serve.app import main
 
 MODE_DIR_BINDING = "/opt/ml/model/"
 _DEFAULT_ENV_VARS = {}
@@ -29,6 +28,8 @@ class InProcessMultiModelServer:
 
     def _start_serving(self):
         """Initializes the start of the server"""
+        from sagemaker.serve.app import main
+
         asyncio.create_task(main())
 
         time.sleep(10)