Prevent logger from being pickled

glemaitre · Jul 4, 2016 · 142077d · 142077d
1 parent 388c202
commit 142077d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/imblearn/base.py b/imblearn/base.py
@@ -225,3 +225,9 @@ def _sample(self, X, y):
             The corresponding label of `X_resampled`
         """
         pass
+
+    def __getstate__(self):
+        """Prevent logger from being pickled"""
+        object_dictionary = dict(self.__dict__)
+        del object_dictionary['logger']
+        return object_dictionary