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

pytorch · Dec 22, 2020 · ca6fdd6 · ca6fdd6
1 parent af5cb00
commit ca6fdd6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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:

diff --git a/version.txt b/version.txt
@@ -0,0 +1 @@
+0.9.0a0