display version info

deepmodeling · Sep 25, 2019 · 70c5398 · 70c5398
1 parent 2c4df8a
commit 70c5398
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/deepmd/__about__.py b/deepmd/__about__.py
@@ -0,0 +1 @@
+__version__ = 'unknown'
diff --git a/setup.py b/setup.py
@@ -38,7 +38,7 @@
 setup(
     name="deepmd-kit",
     setup_requires=setup_requires,
-    use_scm_version={'write_to': 'source/train/_version.py'},
+    use_scm_version={'write_to': 'deepmd/_version.py'},
     author="Han Wang",
     author_email="[email protected]",
     description="A deep learning package for many-body potential energy representation and molecular dynamics",

diff --git a/source/train/__init__.py b/source/train/__init__.py
@@ -6,3 +6,8 @@
 
 set_mkl()
 
+try:
+    from ._version import version as __version__
+except ImportError:
+    from .__about__ import __version__
+