accounts for inferred chr

brwnj · Feb 20, 2020 · 9865032 · 9865032
1 parent 922703a
commit 9865032
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/covviz/ped.py b/covviz/ped.py
@@ -73,9 +73,17 @@ def parse_ped(path, traces, sample_col, sex_chroms, sex_vals="1,2"):
 
             if via_indexcov:
                 # inferred sex
-                ped_data["inferred"]["x"].append(row["CN%s" % sex_chroms[0]])
+                try:
+                    ped_data["inferred"]["x"].append(row["CN%s" % sex_chroms[0]])
+                except KeyError:
+                    ped_data["inferred"]["x"].append(row["CNchr%s" % sex_chroms[0]])
                 try:
                     ped_data["inferred"]["y"].append(row["CN%s" % sex_chroms[1]])
+                except KeyError:
+                    try:
+                        ped_data["inferred"]["y"].append(row["CNchr%s" % sex_chroms[1]])
+                    except IndexError:
+                        ped_data["inferred"]["y"].append(0)
                 except IndexError:
                     ped_data["inferred"]["y"].append(0)