add PEP-396 version attribute (#435)

Author: jesterhazy

setup.py

@@ -12,18 +12,27 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
-from glob import glob
 import os
+import re
+from glob import glob
 
 from setuptools import setup, find_packages
 
 
+def get_version():
+    root = os.path.dirname(__file__)
+    version_re = re.compile(r'''__version__ = ['"]([0-9.]+)['"]''')
+
+    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()
 
 
 setup(name="sagemaker",
-      version="1.11.3",
+      version=get_version(),
       description="Open source library for training and deploying models on Amazon SageMaker.",
       packages=find_packages('src'),
       package_dir={'': 'src'},
@@ -44,7 +53,8 @@ def read(fname):
       ],
 
       # Declare minimal set for installation
-      install_requires=['boto3>=1.4.8', 'numpy>=1.9.0', 'protobuf>=3.1', 'scipy>=0.19.0', 'urllib3 >=1.21, <1.23',
+      install_requires=['boto3>=1.4.8', 'numpy>=1.9.0', 'protobuf>=3.1', 'scipy>=0.19.0',
+                        'urllib3 >=1.21, <1.23',
                         'PyYAML>=3.2', 'protobuf3-to-dict>=0.1.5', 'docker-compose>=1.21.0'],
 
       extras_require={

src/sagemaker/__init__.py

@@ -34,3 +34,5 @@
 from sagemaker.session import production_variant  # noqa: F401
 from sagemaker.session import s3_input  # noqa: F401
 from sagemaker.session import get_execution_role  # noqa: F401
+
+__version__ = '1.11.3'

tests/unit/test_init.py

@@ -0,0 +1,19 @@
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+from __future__ import absolute_import
+
+import sagemaker
+
+
+def test_version():
+    assert sagemaker.__version__