prepush fix

Author: Bryannah Hernandez

src/sagemaker/serve/mode/in_process_mode.py

@@ -5,6 +5,7 @@
 import logging
 from typing import Dict, Type
 import time
+from datetime import datetime, timedelta
 
 from sagemaker.base_predictor import PredictorBase
 from sagemaker.serve.spec.inference_spec import InferenceSpec
@@ -13,7 +14,6 @@
 from sagemaker.serve.utils.exceptions import LocalDeepPingException
 from sagemaker.serve.model_server.multi_model_server.server import InProcessMultiModelServer
 from sagemaker.session import Session
-from datetime import datetime, timedelta
 
 logger = logging.getLogger(__name__)
 

tests/unit/sagemaker/serve/mode/test_in_process_mode.py

@@ -13,7 +13,7 @@
 from __future__ import absolute_import
 
 import unittest
-from unittest.mock import MagicMock, patch, Mock, mock_open
+from unittest.mock import patch, Mock
 
 from sagemaker.serve.mode.in_process_mode import InProcessMode
 from sagemaker.serve import SchemaBuilder