Avoid flakiness in pylint TF things

Author: Ignacio Quintero

src/sagemaker/tensorflow/__init__.py

@@ -22,8 +22,9 @@
 from distutils.version import LooseVersion  # noqa: E402, F401 pylint: disable=no-name-in-module
 import tensorflow  # noqa: E402, F401
 
-if LooseVersion(tensorflow.__version__) < LooseVersion("1.3.0"):
-    message = 'Tensorflow version must be >= 1.3.0. Current version: {}'.format(tensorflow.__version__)
+if LooseVersion(tensorflow.__version__) < LooseVersion("1.3.0"):  # pylint: disable=no-member
+    message = 'Tensorflow version must be >= 1.3.0. Current version: {}'.format(
+        tensorflow.__version__)  # pylint: disable=no-member
     raise AssertionError(message)
 
 from sagemaker.tensorflow.estimator import TensorFlow  # noqa: E402, F401

src/sagemaker/tensorflow/predictor.py

@@ -17,8 +17,8 @@
 import google.protobuf.json_format as json_format
 from google.protobuf.message import DecodeError
 from protobuf_to_dict import protobuf_to_dict
-from tensorflow.core.framework import tensor_pb2
-from tensorflow.python.framework import tensor_util
+from tensorflow.core.framework import tensor_pb2  # pylint: disable=no-name-in-module
+from tensorflow.python.framework import tensor_util  # pylint: disable=no-name-in-module
 
 from sagemaker.content_types import CONTENT_TYPE_JSON, CONTENT_TYPE_OCTET_STREAM, CONTENT_TYPE_CSV
 from sagemaker.predictor import json_serializer, csv_serializer