Update setup.py

miaohancheng · Nov 8, 2024 · a42d5c0 · a42d5c0
1 parent e398612
commit a42d5c0
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -1,4 +1,5 @@
 from setuptools import setup
+import os
 
 dependencies = [
     'catboost>=1.2.7',
@@ -13,15 +14,13 @@
     'imbalanced-learn>=0.12.3'
     ]
 
-VERSION = "0.5"
-
 with open('README.md', 'r') as f:
     long_description = f.read()
 
 setup(
     name='pysmatch',
     packages=['pysmatch'],
-    version=VERSION,
+    version=os.getenv('PACKAGE_VERSION', '0.1'),
     description='Propensity Score Matching(PSM) on python',
     author='Miao HanCheng',
     author_email='hanchengmiao@gmail.com',