diff --git a/setup.py b/setup.py index 41f338ca6..a3be04668 100644 --- a/setup.py +++ b/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) diff --git a/smdebug/_version.py b/smdebug/_version.py index 793c25142..3d187266f 100644 --- a/smdebug/_version.py +++ b/smdebug/_version.py @@ -1 +1 @@ -__version__ = "0.5a" +__version__ = "0.5.0"