From 30535aa6142538e6527556ff3f34bd05df024e06 Mon Sep 17 00:00:00 2001
From: Karim Nakad <knakad@amazon.com>
Date: Tue, 9 Jul 2019 15:20:14 -0700
Subject: [PATCH] change: enable wrong-import-position pylint check

---
 .pylintrc                            | 1 -
 src/sagemaker/amazon/record_pb2.py   | 3 ++-
 src/sagemaker/tensorflow/__init__.py | 9 +++++++--
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/.pylintrc b/.pylintrc
index 2a1c08f880..1e8b52303d 100644
--- a/.pylintrc
+++ b/.pylintrc
@@ -100,7 +100,6 @@ disable=
     simplifiable-if-expression, # TODO: Simplify expressions
     too-many-public-methods, # TODO: Resolve
     ungrouped-imports, # TODO: Group imports
-    wrong-import-position, # TODO: Correct import positions
     consider-using-ternary, # TODO: Consider ternary expressions
     chained-comparison, # TODO: Simplify chained comparison between operands
     simplifiable-if-statement, # TODO: Simplify ifs
diff --git a/src/sagemaker/amazon/record_pb2.py b/src/sagemaker/amazon/record_pb2.py
index cf4578c571..183f807980 100644
--- a/src/sagemaker/amazon/record_pb2.py
+++ b/src/sagemaker/amazon/record_pb2.py
@@ -3,13 +3,14 @@
 
 import sys
 
-_b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode("latin1"))
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
 from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf import descriptor_pb2
 
+_b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode("latin1"))
+
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
diff --git a/src/sagemaker/tensorflow/__init__.py b/src/sagemaker/tensorflow/__init__.py
index 82bfdd9dc8..c521df20fc 100644
--- a/src/sagemaker/tensorflow/__init__.py
+++ b/src/sagemaker/tensorflow/__init__.py
@@ -19,5 +19,10 @@
 # classes for tensorflow serving. Currently tensorflow_serving_api can only be pip-installed for python 2.
 sys.path.append(os.path.dirname(__file__))
 
-from sagemaker.tensorflow.estimator import TensorFlow  # noqa: E402, F401
-from sagemaker.tensorflow.model import TensorFlowModel, TensorFlowPredictor  # noqa: E402, F401
+from sagemaker.tensorflow.estimator import (  # noqa: E402, F401 # pylint: disable=wrong-import-position
+    TensorFlow,
+)
+from sagemaker.tensorflow.model import (  # noqa: E402, F401 # pylint: disable=wrong-import-position
+    TensorFlowModel,
+    TensorFlowPredictor,
+)