Merge pull request #6 from XingerTang/main

Revert the change of writePhase()
AlphaGenes · Jul 17, 2023 · 65456ea · 65456ea
2 parents d62c78a + 8b4e78f
commit 65456ea
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/Pedigree.py b/Pedigree.py
@@ -932,13 +932,9 @@ def writeGenotypes(self, outputFile):
 
     def writePhase(self, outputFile):
         data_list = []
-        for ind in self:
-            if ind.haplotypes.ndim == 2:  # diploid
-                data_list.append((ind.idx, ind.haplotypes[0]))
-                data_list.append((ind.idx, ind.haplotypes[1]))
-            elif ind.haplotypes.ndim == 1:  # haploid
-                data_list.append((ind.idx, ind.haplotypes))
-                data_list.append((ind.idx, ind.haplotypes))
+        for ind in self :
+            data_list.append( (ind.idx, ind.haplotypes[0]) )
+            data_list.append( (ind.idx, ind.haplotypes[1]) )
 
         MultiThreadIO.writeLines(outputFile, data_list, str)