From 2ecfa81e52aa8b957166648073a06c516c8a88f4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=C3=A1t=C3=A9=20Balajti?= <mate.balajti@unibas.ch>
Date: Wed, 9 Aug 2023 13:44:45 +0200
Subject: [PATCH] fix: fragmentation function

---
 term_frag_sel/fragmentation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/term_frag_sel/fragmentation.py b/term_frag_sel/fragmentation.py
index f3471a0..9cd8a0f 100644
--- a/term_frag_sel/fragmentation.py
+++ b/term_frag_sel/fragmentation.py
@@ -24,7 +24,7 @@ def fragmentation(fasta: dict, seq_counts: pd.DataFrame,
 
     term_frags = []
     for seq_id, seq in fasta.items():
-        counts = seq_counts[seq_counts["seqID"] == str(seq_id)]["count"]
+        counts = seq_counts[seq_counts["seqID"] == seq_id]["count"]
         for _ in range(counts.iloc[0]):
             cuts = []
             seq_len = len(seq)
-- 
GitLab