Add git SHA to wheel versions using setuptools_scm

Signed-off-by: jbernloehr <[email protected]>

Author: janbernloehr

pyproject.toml

@@ -42,16 +42,32 @@ dev = [
     "expecttest", # test_tokenizer
 ]
 
-[tool.setuptools.dynamic]
-version = {file = "assets/version.txt"}
+[tool.setuptools_scm]
+# Automatically generate version from git tags and commits at BUILD TIME
+# Development builds: 0.2.0.post77+gf5e3a84e (77 commits after v0.2.0 tag)
+# Tagged releases: 0.2.0 (when building from a git tag)
+# To override version for releases: set SETUPTOOLS_SCM_PRETEND_VERSION=x.y.z
+# This writes the version to torchtitan/_version.py during wheel build
+version_file = "torchtitan/_version.py"
+version_file_template = """\
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+# This file is generated by setuptools_scm at build time
+# DO NOT EDIT - it will be overwritten during the build process
+__version__ = "{version}"
+"""
+tag_regex = "^(?P<prefix>v)?(?P<version>[^\\+]+)(?P<suffix>.*)?$"
 
 [tool.codespell]
 skip = ["*.json"]
 ignore-words-list = ["assertin", "datas" ,"indx", "inpt", "nd", "socio-economic"]
 
 # ---- Explicit project build information ---- #
 [build-system]
-requires = ["setuptools>=61.0"]
+requires = ["setuptools>=61.0", "setuptools-scm>=8.0"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools.packages.find]

torchtitan/__init__.py

@@ -6,3 +6,6 @@
 
 # Import to register quantization modules.
 import torchtitan.components.quantization  # noqa: F401
+from torchtitan._version import __version__
+
+__all__ = ["__version__"]

torchtitan/_version.py

@@ -0,0 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+# This file is generated by setuptools_scm at build time
+# DO NOT EDIT - it will be overwritten during the build process
+__version__ = "0.2.0+unknown"