Closes #318.

.gitignore

@@ -23,6 +23,7 @@ share/python-wheels/
 .installed.cfg
 *.egg
 MANIFEST
+src/team_comm_tools/_version.py
 
 # IntelliJ
 .idea/

MANIFEST.in

@@ -2,4 +2,5 @@ recursive-include src/team_comm_tools/features/lexicons *
 include src/mypackage/features/assets/*
 include README.md
 include LICENSE
-include requirements.txt
+include requirements.txt
+include pyproject.toml

pyproject.toml

@@ -1,6 +1,6 @@
 [build-system]
 requires = [
-  "setuptools >= 42", "wheel"
+  "setuptools >= 42", "wheel", "toml"
 ]
 build-backend = "setuptools.build_meta"
 
@@ -13,6 +13,7 @@ dependencies = [
   "convokit==3.0.0",
   "emoji==1.7.0",
   "flask==3.0.3",
+  "fuzzywuzzy==0.18.0",
   "gensim>=4.3.3",
   "matplotlib>=3.0.0",
   "nltk==3.9.1",
@@ -39,8 +40,7 @@ dependencies = [
   "transformers==4.44.0",
   "tqdm>=4.66.5",
   "tzdata>=2023.3",
-  "tzlocal==5.2",
-  "fuzzywuzzy==0.18.0"
+  "tzlocal==5.2"
 ]
 authors = [
   {name = "Xinlan Emily Hu", email = "[email protected]"},

requirements.txt

@@ -2,6 +2,7 @@ chardet>=3.0.4
 convokit==3.0.0
 emoji==1.7.0
 flask==3.0.3
+fuzzywuzzy==0.18.0
 gensim>=4.3.3
 matplotlib>=3.0.0
 nltk==3.9.1
@@ -28,5 +29,4 @@ torchvision==0.19.1
 transformers==4.44.0
 tqdm>=4.66.5
 tzdata>=2023.3
-tzlocal==5.2
-fuzzywuzzy==0.18.0
+tzlocal==5.2

setup.py

@@ -0,0 +1,31 @@
+from setuptools import setup
+from setuptools.command.build_py import build_py
+import toml
+import os
+
+def get_version_from_pyproject():
+    with open("pyproject.toml", "r") as f:
+        pyproject_data = toml.load(f)
+    return pyproject_data["project"]["version"]
+
+version = get_version_from_pyproject()
+version_file = os.path.join(os.path.dirname(__file__), "src/team_comm_tools", "_version.py")
+if not os.path.exists(version_file):
+    with open(version_file, "w") as f:
+        f.write(f'__version__ = "{version}"\n')
+
+class BuildPyCommand(build_py):
+    def run(self):
+        # Generate _version.py before building, which helps automatically update the .__version__ parameter
+        version = get_version_from_pyproject()
+        version_path = os.path.join(self.build_lib, "src/team_comm_tools", "_version.py")
+        os.makedirs(os.path.dirname(version_path), exist_ok=True)
+        with open(version_path, "w") as f:
+            f.write(f'__version__ = "{version}"\n')
+        super().run()
+
+setup(
+    cmdclass={
+        'build_py': BuildPyCommand,
+    }
+)

src/team_comm_tools/__init__.py

@@ -1 +1,2 @@
-from .feature_builder import FeatureBuilder
+from .feature_builder import FeatureBuilder
+from ._version import __version__