Reformat

astheeggeggs · Jun 1, 2024 · ca0a0f3 · ca0a0f3
1 parent 6b22309
commit ca0a0f3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_non_tree.py b/tests/test_non_tree.py
@@ -398,9 +398,7 @@ def verify(self, ts, scale_mutation_rate, include_ancestors):
             G_vs = core.convert_haplotypes_to_phased_genotypes(H_vs)
             s = core.convert_haplotypes_to_unphased_genotypes(query)
 
-            V_vs, P_vs, ll_vs = vd.forwards_viterbi_dip_low_mem(
-                n, m, G_vs, s, e_vs, r
-            )
+            V_vs, P_vs, ll_vs = vd.forwards_viterbi_dip_low_mem(n, m, G_vs, s, e_vs, r)
             path_vs = vd.backwards_viterbi_dip(m, V_vs, P_vs)
             phased_path_vs = vd.get_phased_path(n, path_vs)
             path_ll_vs = vd.path_ll_dip(n, m, G_vs, phased_path_vs, s, e_vs, r)
@@ -430,7 +428,9 @@ def verify(self, ts, scale_mutation_rate, include_ancestors):
             self.assertAllClose(ll_vs, ll_tmp)
 
             if not include_ancestors:
-                V_tmp, P_tmp, ll_tmp = vd.forwards_viterbi_dip_naive(n, m, G_vs, s, e_vs, r)
+                V_tmp, P_tmp, ll_tmp = vd.forwards_viterbi_dip_naive(
+                    n, m, G_vs, s, e_vs, r
+                )
                 path_tmp = vd.backwards_viterbi_dip(m, V_tmp[m - 1, :, :], P_tmp)
                 phased_path_tmp = vd.get_phased_path(n, path_tmp)
                 path_ll_tmp = vd.path_ll_dip(n, m, G_vs, phased_path_tmp, s, e_vs, r)