setup for prerelease builds (#763)

bdcba9c8 · Quan (Andy) Gan · GitHub · 2ee3c78c · bdcba9c8 · bdcba9c8
Unverified Commit bdcba9c8 authored Aug 14, 2019 by Quan (Andy) Gan Committed by GitHub Aug 14, 2019
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

python/setup.py python/setup.py +1 -1

python/update_version.py python/update_version.py +2 -1

No files found.
--- a/python/setup.py
+++ b/python/setup.py
@@ -110,7 +110,7 @@ if include_libs:
    }

 setup(
-    name='dgl',
+    name='dgl' + os.getenv('DGL_PACKAGE_SUFFIX', ''),
    version=VERSION,
    description='Deep Graph Library',
    zip_safe=False,

--- a/python/update_version.py
+++ b/python/update_version.py
@@ -11,7 +11,8 @@ import re
 # current version
 # We use the version of the incoming release for code
 # that is under development
-__version__ = "0.3"
+__version__ = "0.3" + os.getenv('DGL_PRERELEASE', '')
+print(__version__)

 # Implementations
 def update(file_name, pattern, repl):