Move version definition back to setup.py (#612)

Author: laurenyu

doc/conf.py

@@ -13,6 +13,7 @@
 from __future__ import absolute_import
 
 import os
+import pkg_resources
 import sys
 from datetime import datetime
 from unittest.mock import MagicMock
@@ -32,8 +33,8 @@ def __getattr__(cls, name):
                 'numpy', 'scipy', 'scipy.sparse']
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
 
-version = '1.18.1'
 project = u'sagemaker'
+version = pkg_resources.require(project)[0].version
 
 # Add any Sphinx extension module names here, as strings. They can be extensions
 # coming with Sphinx (named 'sphinx.ext.*') or your custom ones.

setup.py

@@ -13,21 +13,12 @@
 from __future__ import absolute_import
 
 import os
-import re
 from glob import glob
 import sys
 
 from setuptools import setup, find_packages
 
 
-def get_version():
-    root = os.path.dirname(__file__)
-    version_re = re.compile(r'''__version__ = ['"]([0-9.]+((post|dev)\d+)?)['"]''')
-
-    init = read(os.path.join(root, 'src/sagemaker', '__init__.py'))
-    return version_re.search(init).group(1)
-
-
 def read(fname):
     return open(os.path.join(os.path.dirname(__file__), fname)).read()
 
@@ -42,7 +33,7 @@ def read(fname):
     required_packages.append('enum34>=1.1.6')
 
 setup(name="sagemaker",
-      version=get_version(),
+      version='1.18.1',
       description="Open source library for training and deploying models on Amazon SageMaker.",
       packages=find_packages('src'),
       package_dir={'': 'src'},

src/sagemaker/__init__.py

@@ -12,6 +12,8 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
+import pkg_resources
+
 from sagemaker import estimator, parameter, tuner  # noqa: F401
 from sagemaker.amazon.kmeans import KMeans, KMeansModel, KMeansPredictor  # noqa: F401
 from sagemaker.amazon.pca import PCA, PCAModel, PCAPredictor  # noqa: F401
@@ -39,4 +41,4 @@
 from sagemaker.session import s3_input  # noqa: F401
 from sagemaker.session import get_execution_role  # noqa: F401
 
-__version__ = '1.18.1'
+__version__ = pkg_resources.require('sagemaker')[0].version