update suffix (#5128)

c6228c01 · Quan (Andy) Gan · GitHub · b7042a14 · c6228c01
Unverified Commit c6228c01 authored Jan 10, 2023 by Quan (Andy) Gan Committed by GitHub Jan 10, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

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

No files found.
--- a/python/update_version.py
+++ b/python/update_version.py
@@ -12,7 +12,12 @@ import re
 # current version
 # We use the version of the incoming release for code
 # that is under development
-__version__ = "1.0" + os.getenv("DGL_PRERELEASE", "")
+# The environment variable DGL_PRERELEASE is the prerelase suffix
+# (usually "aYYMMDD")
+# The environment variable DGL_VERSION_SUFFIX is the local version label
+# suffix for indicating CPU and CUDA versions as in PEP 440 (e.g. "+cu102")
+__version__ = "1.0" + os.getenv("DGL_PRERELEASE", "") 
+__verison__ += os.getenv("DGL_VERSION_SUFFIX", "")
 print(__version__)

 # Implementations