truvari anno trf compatible with adotto v1.2.1

Looks for 'motif' and 'repeat' key from json
ACEnglish · Oct 24, 2024 · 0817fab · 0817fab
1 parent a1521c1
commit 0817fab
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/repo_utils/test_files/beds/repeats.adotto.bed.gz b/repo_utils/test_files/beds/repeats.adotto.bed.gz
diff --git a/truvari/annotations/trf.py b/truvari/annotations/trf.py
@@ -64,7 +64,7 @@ def __init__(self, region, reference, motif_similarity=0.90, buf=5):
         self.region = region
         self.reference = reference
         self.motif_similarity = motif_similarity
-        self.known_motifs = {_["repeat"]: _["copies"]
+        self.known_motifs = {_["repeat"] if "repeat" in _ else _["motif"]: _["copies"]
                              for _ in self.region["annos"]}
         self.buffer = buf