fix error in setup.py

lartpang · Mar 31, 2021 · f5ce652 · f5ce652
1 parent db9bd84
commit f5ce652
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/py_sod_metrics/sod_metrics.py b/py_sod_metrics/sod_metrics.py
@@ -5,7 +5,6 @@
 
 _EPS = 1e-16
 _TYPE = np.float64
-__version__ = "1.2.4"
 
 
 def _prepare_data(pred: np.ndarray, gt: np.ndarray) -> tuple:

diff --git a/setup.py b/setup.py
@@ -2,15 +2,16 @@
 
 from setuptools import find_packages, setup
 
-from py_sod_metrics.sod_metrics import __version__
+with open("./version.txt", encoding="utf-8", mode="r") as f:
+    version = f.readline().strip()
 
 with open("readme.md", "r") as fh:
     long_description = fh.read()
 
 setup(
     name="pysodmetrics",
     packages=find_packages(),
-    version=__version__,
+    version=version,
     license="MIT",
     description="A simple and efficient implementation of SOD metrics.",
     long_description=long_description,

diff --git a/version.txt b/version.txt
@@ -0,0 +1 @@
+1.2.4