Make imbalanced-learn optional

DataCanvasIO · Feb 18, 2024 · 171c549 · 171c549
1 parent aa44d23
commit 171c549
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/hypergbm/hyper_gbm.py b/hypergbm/hyper_gbm.py
@@ -43,14 +43,14 @@
     from imblearn.over_sampling import RandomOverSampler, SMOTE, ADASYN
     from imblearn.under_sampling import RandomUnderSampler, NearMiss, TomekLinks, EditedNearestNeighbours
 
-    imblearn_istalled = True
+    imblearn_installed = True
 except:
     logger.warning('Failed to load imbalanced-learn', exc_info=sys.exc_info())
-    imblearn_istalled = False
+    imblearn_installed = False
 
 
 def get_sampler(sampler):
-    if imblearn_istalled:
+    if imblearn_installed:
         samplers = {'RandomOverSampler': RandomOverSampler,
                     'SMOTE': SMOTE,
                     'ADASYN': ADASYN,