Commit 35341c80 authored by Tiago Peixoto's avatar Tiago Peixoto
Browse files

Add missing conversion protection in CovariateBlockState.__setstate__()

parent 577958b5
Pipeline #42 passed with stage
......@@ -23,7 +23,8 @@ import sys
if sys.version_info < (3,):
range = xrange
from .. import _degree, _prop, Graph, GraphView, libcore, _get_rng, PropertyMap
from .. import _degree, _prop, Graph, GraphView, libcore, _get_rng, PropertyMap, \
conv_pickle_state
import random
from numpy import *
import numpy
......@@ -465,6 +466,7 @@ class CovariateBlockState(BlockState):
return state
def __setstate__(self, state):
conv_pickle_state(state)
self.__init__(**state)
return state
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment