Commit f32baccc authored by Tiago Peixoto's avatar Tiago Peixoto

inference.uncertain_blockmodel: Fix pickling bug

parent 8efb5176
......@@ -88,8 +88,8 @@ struct Measured
(source(e, _u) == target(e, _u))))
continue;
auto ge = get_edge<false>(source(e, _g), target(e, _g));
_T += (ge != _null_edge) ? _x[e] : _x_default;
_M += (ge != _null_edge) ? _n[e] : _n_default;
_T += (ge != _null_edge) ? _x[ge] : _x_default;
_M += (ge != _null_edge) ? _n[ge] : _n_default;
}
size_t N = num_vertices(_g);
......
......@@ -366,6 +366,7 @@ class MeasuredBlockState(UncertainBaseState):
x_default=self.x_default,
fp_params=dict(alpha=self.alpha, beta=self.beta),
fn_params=dict(mu=self.mu, nu=self.nu), phi=self.phi,
nested=self.nbstate is not None,
bstate=(self.nbstate if self.nbstate is not None
else self.bstate), self_loops=self.self_loops)
......
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