[fix] wrong axis for qC diploid init

src/variational_distributions/var_dists.py

@@ -262,7 +262,7 @@ def _init_diploid(self, nodes: list, skewness=5.):
         root_startprob = torch.ones(self.config.n_states)
         root_startprob[2] = skewness
         root_transmat = torch.ones((self.config.n_states, self.config.n_states))
-        root_transmat[2, :] = skewness
+        root_transmat[:, 2] = skewness
         # normalize and log-transform
         self.eta1[nodes, :] = root_startprob - torch.logsumexp(root_startprob, dim=-1, keepdim=True)
         normalized_log_transmat = root_transmat - torch.logsumexp(root_transmat, dim=-1, keepdim=True)