Change the unique checking

Author: Guillaume Lemaitre

imblearn/base.py

@@ -81,14 +81,11 @@ def fit(self, X, y):
 
         self.logger.info('Compute classes statistics ...')
 
-        # Get all the unique elements in the target array
-        uniques = np.unique(y)
-
         # # Raise an error if there is only one class
         # if uniques.size == 1:
         #     raise RuntimeError("Only one class detected, aborting...")
         # Raise a warning for the moment to be compatible with BaseEstimator
-        if uniques.size == 1:
+        if np.unique(y).size == 1:
             warnings.warn('Only one class detected, something will get wrong',
                           RuntimeWarning)
 
@@ -103,7 +100,7 @@ def fit(self, X, y):
         self.min_c_ = min(self.stats_c_, key=self.stats_c_.get)
         self.maj_c_ = max(self.stats_c_, key=self.stats_c_.get)
 
-        self.logger.info('%s classes detected: %s', uniques.size,
+        self.logger.info('%s classes detected: %s', np.unique(y).size,
                          self.stats_c_)
 
         # Check if the ratio provided at initialisation make sense