From 8a648f677f788ea2778a99502cb576470361198b Mon Sep 17 00:00:00 2001
From: Matt Wala <wala1@illinois.edu>
Date: Fri, 17 Feb 2017 16:39:41 -0600
Subject: [PATCH] Fix pickleability in Py2.
---
loopy/tools.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/loopy/tools.py b/loopy/tools.py
index 36f015a29..524d2cf72 100644
--- a/loopy/tools.py
+++ b/loopy/tools.py
@@ -348,8 +348,6 @@ class _PickledObjectWrapper(object):
A class meant to wrap a pickled value (for :class:`LazilyUnpicklingDictionary`).
"""
- __slots__ = ["objstring"]
-
@classmethod
def from_object(cls, obj):
if isinstance(obj, cls):
@@ -364,6 +362,9 @@ class _PickledObjectWrapper(object):
from pickle import loads
return loads(self.objstring)
+ def __getstate__(self):
+ return {"objstring": self.objstring}
+
import collections
--
GitLab