Get logger in fit and sample using private method

Author: chkoar

imblearn/base.py

@@ -47,7 +47,7 @@ def __init__(self, ratio='auto'):
         """
 
         self.ratio = ratio
-        self.logger = logging.getLogger(__name__)
+        
 
     def fit(self, X, y):
         """Find the classes statistics before to perform sampling.
@@ -69,6 +69,8 @@ def fit(self, X, y):
 
         # Check the consistency of X and y
         X, y = check_X_y(X, y)
+		
+        self._get_logger()
 
         self.min_c_ = None
         self.maj_c_ = None
@@ -138,6 +140,8 @@ def sample(self, X, y):
 
         # Check the consistency of X and y
         X, y = check_X_y(X, y)
+		
+        self._get_logger()
 
         # Check that the data have been fitted
         if not hasattr(self, 'stats_c_'):
@@ -220,3 +224,7 @@ def __getstate__(self):
         object_dictionary = self.__dict__.copy()
         del object_dictionary['logger']
         return object_dictionary
+
+		
+    def _get_logger(self):
+        self.logger = logging.getLogger(__name__)