Fix flake8 failures

saimidu · saimidu · commit bf0e407d6840 · 2021-10-15T12:15:32.000-07:00
diff --git a/src/sagemaker_pytorch_serving_container/default_pytorch_inference_handler.py b/src/sagemaker_pytorch_serving_container/default_pytorch_inference_handler.py
@@ -13,7 +13,6 @@
 from __future__ import absolute_import
 
 import os
-import textwrap
 
 import torch
 from sagemaker_inference import (
diff --git a/test/integration/sagemaker/test_mnist_default_inference.py b/test/integration/sagemaker/test_mnist_default_inference.py
@@ -21,12 +21,6 @@
 from sagemaker.pytorch import PyTorchModel, PyTorchPredictor
 
 from integration import (
-    model_cpu_tar,
-    model_gpu_tar,
-    mnist_cpu_script,
-    mnist_gpu_script,
-    model_eia_tar,
-    mnist_eia_script,
     default_model_script,
     default_model_tar,
     default_model_eia_script,
