Skip to content

Commit be06d8d

Browse files
committed
Pickle fix for SortedSet on py27
1 parent 7d3160d commit be06d8d

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

sortedcontainers/sortedset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -702,7 +702,7 @@ def __reduce__(self):
702702
confuse pickle so customize the reducer.
703703
704704
"""
705-
return (type(self)._fromset, (self._set, self._key))
705+
return (type(self), (self._set, self._key))
706706

707707

708708
@recursive_repr()

tests/test_coverage_sortedset.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -473,6 +473,7 @@ def test_pickle():
473473
import pickle
474474
alpha = SortedSet(range(10000), key=negate)
475475
alpha._reset(500)
476-
beta = pickle.loads(pickle.dumps(alpha))
476+
data = pickle.dumps(alpha)
477+
beta = pickle.loads(data)
477478
assert alpha == beta
478479
assert alpha._key == beta._key

0 commit comments

Comments
 (0)