From ca88bc95716adbcf1d959377a1d00315bc639aa9 Mon Sep 17 00:00:00 2001 From: Yu Date: Mon, 26 Nov 2018 19:54:05 -0800 Subject: [PATCH] Remove unnecessary dependency tensorflow --- CHANGELOG.rst | 5 +++++ src/sagemaker/tensorflow/__init__.py | 8 -------- 2 files changed, 5 insertions(+), 8 deletions(-) diff --git a/CHANGELOG.rst b/CHANGELOG.rst index de123b9d17..28d2e760d9 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -2,6 +2,11 @@ CHANGELOG ========= +1.15.3.dev +========== + +* bug-fix: Remove unnecessary dependency tensorflow + 1.15.2 ====== diff --git a/src/sagemaker/tensorflow/__init__.py b/src/sagemaker/tensorflow/__init__.py index cc61966bff..82bfdd9dc8 100644 --- a/src/sagemaker/tensorflow/__init__.py +++ b/src/sagemaker/tensorflow/__init__.py @@ -19,13 +19,5 @@ # classes for tensorflow serving. Currently tensorflow_serving_api can only be pip-installed for python 2. sys.path.append(os.path.dirname(__file__)) -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"): # 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 from sagemaker.tensorflow.model import TensorFlowModel, TensorFlowPredictor # noqa: E402, F401