Fix version imports and versioningit setup.

Fix version and add test
choderalab · Aug 5, 2024 · 0e06eda · 0e06eda
2 parents c18b034 + 3a04c55
commit 0e06eda
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 105 deletions.
diff --git a/mtenn/__init__.py b/mtenn/__init__.py
@@ -1 +1,4 @@
 """Modular Training and Evaluation of Neural Networks"""
+
+from importlib.metadata import version
+__version__ = version("mtenn")
diff --git a/mtenn/_version.py b/mtenn/_version.py
diff --git a/mtenn/_version_git.py b/mtenn/_version_git.py
diff --git a/mtenn/tests/test_version.py b/mtenn/tests/test_version.py
@@ -0,0 +1,4 @@
+import mtenn
+
+def test_version():
+    assert mtenn.__version__
diff --git a/pyproject.toml b/pyproject.toml
@@ -33,11 +33,6 @@ dirty = "{base_version}+{distance}.{vcs}{rev}.dirty"
 distance-dirty = "{base_version}+{distance}.{vcs}{rev}.dirty"
 
 [tool.versioningit.vcs]
-# The method key:
-method = "git"  # <- The method name
-# Parameters to pass to the method:
+method = "git" 
 match = ["*"]
-default-tag = "0.0.0"
-
-[tool.versioningit.write]
-file = "mtenn/_version.py"
+default-tag = "0.0.0"