Move package version to setup.cfg

4 files changed, 5 insertions, 8 deletions

diff --git a/bicleaner_ai/__init__.py b/bicleaner_ai/__init__.py
index 14c7c45..e3328c9 100755
--- a/bicleaner_ai/__init__.py
+++ b/bicleaner_ai/__init__.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python
 
 name = "bicleaner_ai"
-#__all__=["bicleaner_hardrules", "bicleaner_classifier_full", "bicleaner_train"]
-#__all__=["classify", "train"]
+__version__ = "2.0"
 
 from .util import  *
diff --git a/bicleaner_ai/classify.py b/bicleaner_ai/classify.py
index ea256e6..b4fe132 100644
--- a/bicleaner_ai/classify.py
+++ b/bicleaner_ai/classify.py
@@ -14,15 +14,12 @@ import gc
 
 #Allows to load modules while inside or outside the package
 try:
+    from . import __version__
     from .util import check_positive, check_positive_or_zero, check_positive_between_zero_and_one, logging_setup, get_model
 except (ImportError, SystemError):
+    from bicleaner_ai import __version__
     from util import check_positive, check_positive_or_zero, check_positive_between_zero_and_one, logging_setup, get_model
 
-__author__ = "Jaume Zaragoza"
-__version__ = "Version 1.0 # 14/06/2021 #"
-__version__ = "Version 1.0.1 # 16/06/2021 #"
-__version__ = "Version 2.0"
-
 
 # Create an argument parser and add all the arguments
 def argument_parser():
diff --git a/setup.cfg b/setup.cfg
new file mode 100644
index 0000000..4cc6065
--- /dev/null
+++ b/setup.cfg
@@ -0,0 +1,2 @@
+[metadata]
+version = attr: bicleaner_ai.__version__
diff --git a/setup.py b/setup.py
index 21508fd..34ebd6c 100755
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,6 @@ with open("requirements.txt") as rf:
     
 setuptools.setup(
     name="bicleaner-ai",
-    version="2.0",
     install_requires=requirements,
     license="GNU General Public License v3.0",
     author="Prompsit Language Engineering",