Commit 65ea9489 authored by Tiago Peixoto's avatar Tiago Peixoto
Browse files Fix copying of NestedBlockState with sampling == True

parent df948de9
...@@ -79,8 +79,11 @@ class NestedBlockState(object): ...@@ -79,8 +79,11 @@ class NestedBlockState(object):
if sampling: if sampling:
self.hstate_args = dict(self.hstate_args, vweight="nonempty", self.hstate_args = dict(self.hstate_args, vweight="nonempty",
copy_bg=False, B=g.num_vertices()) copy_bg=False, B=g.num_vertices())
self.state_args = dict(self.state_args, vweight="unity", eweight="unity", self.state_args = dict(self.state_args, B=g.num_vertices())
B=g.num_vertices()) if self.state_args.get("vweight", None) is None:
self.state_args["vweight"] = "unity"
if self.state_args.get("eweight", None) is None:
self.state_args["eweight"] = "unity"
nbs = [] nbs = []
for b in bs: for b in bs:
nb = numpy.zeros(g.num_vertices(), dtype="int") nb = numpy.zeros(g.num_vertices(), dtype="int")
Markdown is supported
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