Fix pickling error

This commit is contained in:
Matthew Tretter 2013-01-29 02:27:03 -05:00
parent 54ca5da15d
commit f0dbe32f7a

View file

@ -47,6 +47,12 @@ class StrategyWrapper(LazyObject):
strategy = strategy()
self._wrapped = strategy
def __getstate__(self):
return {'_wrapped': self._wrapped}
def __setstate__(self, state):
self._wrapped = state['_wrapped']
def __unicode__(self):
return unicode(self._wrapped)