Nightly Build setup

setup.py

@@ -1,13 +1,30 @@
+#!/usr/bin/env python
+""" Amazon SageMaker Debugger is an offering from AWS which helps you automate the debugging of machine learning training jobs.
+This library powers Amazon SageMaker Debugger, and helps you develop better, faster and cheaper models by catching common errors quickly.
+It allows you to save tensors from training jobs and makes these tensors available for analysis, all through a flexible and powerful API.
+It supports TensorFlow, PyTorch, MXNet, and XGBoost on Python 3.6+.
+- Zero Script Change experience on SageMaker when using supported versions of SageMaker Framework containers or AWS Deep Learning containers
+- Full visibility into any tensor which is part of the training process
+- Real-time training job monitoring through Rules
+- Automated anomaly detection and state assertions
+- Interactive exploration of saved tensors
+- Distributed training support
+- TensorBoard support
+
+"""
+
 # Standard Library
 import os
 import sys
+from datetime import date
 
 # Third Party
 import setuptools
 
-exec(open("smdebug/_version.py").read())
+# First Party
+import smdebug
 
-CURRENT_VERSION = __version__
+DOCLINES = (__doc__ or "").split("\n")
 FRAMEWORKS = ["tensorflow", "pytorch", "mxnet", "xgboost"]
 TESTS_PACKAGES = ["pytest", "torchvision", "pandas"]
 INSTALL_REQUIRES = ["protobuf>=3.6.0", "numpy", "packaging", "boto3>=1.10.32"]
@@ -29,8 +46,10 @@ def build_package(version):
     setuptools.setup(
         name="smdebug",
         version=version,
+        long_description="\n".join(DOCLINES[1:]),
+        long_description_content_type="text/x-rst",
         author="AWS DeepLearning Team",
-        description="Automated debugging for machine learning",
+        description=DOCLINES[0],
         url="https://github.com/awslabs/sagemaker-debugger",
         packages=packages,
         classifiers=[
@@ -82,4 +101,14 @@ def git(*args):
 
     sys.exit(1)
 
-build_package(version=CURRENT_VERSION)
+
+def detect_smdebug_version():
+    if "--release" in sys.argv:
+        sys.argv.remove("--release")
+        return smdebug.__version__.strip()
+
+    return smdebug.__version__.strip() + "b" + str(date.today()).replace("-", "")
+
+
+version = detect_smdebug_version()
+build_package(version=version)

smdebug/_version.py

@@ -1 +1 @@
-__version__ = "0.5a"
+__version__ = "0.5.0"