From f04308caaeebe6d5f7921973ae5f9b8d885e3826 Mon Sep 17 00:00:00 2001 From: Alex Rothberg Date: Tue, 6 Jan 2015 14:58:26 -0500 Subject: [PATCH] Fix pickling --- nolearn/lasagne.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/nolearn/lasagne.py b/nolearn/lasagne.py index 35ee029..3618c71 100644 --- 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