Move version definition from setup.py to version.txt (#3202)

ca6fdd6d · Vasilis Vryniotis · GitHub · af5cb00c · ca6fdd6d · ca6fdd6d
Unverified Commit ca6fdd6d authored Dec 22, 2020 by Vasilis Vryniotis Committed by GitHub Dec 22, 2020
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 3 deletions

setup.py setup.py +5 -3

version.txt version.txt +1 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -29,12 +29,14 @@ def get_dist(pkgname):
        return None


-version = '0.9.0a0'
+cwd = os.path.dirname(os.path.abspath(__file__))
+
+version_txt = os.path.join(cwd, 'version.txt')
+with open(version_txt, 'r') as f:
+    version = f.readline().strip()
 sha = 'Unknown'
 package_name = 'torchvision'

-cwd = os.path.dirname(os.path.abspath(__file__))
-
 try:
    sha = subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=cwd).decode('ascii').strip()
 except Exception:

--- a/version.txt
+++ b/version.txt
+0.9.0a0
\ No newline at end of file