Fix pickling

cancan101 · Jan 9, 2015 · f04308c · f04308c
1 parent 3f80b31
commit f04308c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/nolearn/lasagne.py b/nolearn/lasagne.py
@@ -129,6 +129,14 @@ def __init__(
                 "Use 'batch_iterator_train' and 'batch_iterator_test' instead."
                 )
 
+    def __getstate__(self):
+        ret = dict(self.__dict__)
+        del ret['train_iter_']
+        del ret['eval_iter_']
+        del ret['predict_iter_']
+
+        return ret
+
     def _initialize(self):
         if (getattr(self, 'train_iter_', None) is not None
             and getattr(self, 'eval_iter_', None) is not None