Don't understand Wenzhe's code/comments in __sample_z_for_each_edge. MCMC compatibility bug still in.