fix(mi): locus dict contig logic

davidlougheed · Oct 21, 2024 · 91a2bba · 91a2bba
1 parent 846ace5
commit 91a2bba
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/strkit/mi/base.py b/strkit/mi/base.py
@@ -130,9 +130,7 @@ def get_trio_contigs(self, include_sex_chromosomes: bool = False) -> set:
 
         if self._loci_dict:
             # Limit contig set to only contigs which are in the locus dictionary if one is specified.
-            print(contig_set)
-            contig_set = contig_set.intersection({k[0] for k in self._loci_dict})
-            print({k[0] for k in self._loci_dict})
+            contig_set = contig_set.intersection(self._loci_dict.keys())
 
         self._logger.debug("Got %d intersection trio contigs", len(contig_set))