[Versioning] Write version all cases in setup.py (pytorch#1579)

ai4co · Sep 27, 2023 · cbece6b · cbece6b
1 parent 7f42576
commit cbece6b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -174,10 +174,11 @@ def _main(argv):
     if is_nightly:
         version = get_nightly_version()
         write_version_file(version)
-        print("Building wheel {}-{}".format(package_name, version))
-        print(f"BUILD_VERSION is {os.getenv('BUILD_VERSION')}")
     else:
         version = get_version()
+        write_version_file(version)
+    print("Building wheel {}-{}".format(package_name, version))
+    print(f"BUILD_VERSION is {os.getenv('BUILD_VERSION')}")
 
     pytorch_package_dep = _get_pytorch_version(is_nightly)
     print("-- PyTorch dependency:", pytorch_package_dep)