Fix dependencies version conflicts

DataCanvasIO · Oct 18, 2021 · 2658a14 · 2658a14
1 parent d0adac1
commit 2658a14
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/hypergbm/_version.py b/hypergbm/_version.py
@@ -1 +1 @@
-__version__ = '0.2.3'
+__version__ = '0.2.3.1'
diff --git a/hypergbm/estimators.py b/hypergbm/estimators.py
@@ -3,6 +3,8 @@
 
 """
 import sys
+from distutils.version import LooseVersion
+
 import catboost
 import lightgbm
 import numpy as np
@@ -583,7 +585,8 @@ def iteration_scores(self):
     def build_discriminator_callback(self, discriminator):
         if discriminator is None:
             return None
-        if int(catboost.__version__.split('.')[1]) >= 26:
+        # if int(catboost.__version__.split('.')[1]) >= 26:
+        if LooseVersion(catboost.__version__) >= LooseVersion('0.26'):
             callback = CatboostDiscriminationCallback(discriminator=discriminator, group_id=self.group_id)
             self.discriminator_callback = callback
             return callback

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,5 @@
+featuretools>=0.23.0,<1.0
+scikit-learn>=0.22.1,<1.0
 hypernets==0.2.3
 fsspec>=0.8.0
 dask