fix versioning

mattmilten · Dec 22, 2021 · 055fa52 · 055fa52
1 parent b5e240e
commit 055fa52
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 
 setup(
     name="treed",
-    version="2.0.1",
+    version=version,
     author="Matthias Miltenberger",
     author_email="[email protected]",
     description="3D Visualization of Branch-and-Cut Trees using PySCIPOpt",

diff --git a/src/treed/__main__.py b/src/treed/__main__.py
@@ -14,7 +14,8 @@
 parser.add_argument(
     "--version",
     "-v",
-    action="store_true",
+    action="version",
+    version=f"TreeD {__version__}",
     help="show version",
 )
 parser.add_argument(