FIX FIMO lint issue

jmschrei · Feb 16, 2024 · 45d72f1 · 45d72f1
1 parent df8d22b
commit 45d72f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tangermeme/tools/fimo.py b/tangermeme/tools/fimo.py
@@ -301,7 +301,7 @@ def hits(self, X, X_attr=None, threshold=0.0001, batch_size=256, dim=0,
 		hit_idxs = torch.where(scores > score_thresh)        
 		for idxs in tqdm(zip(*hit_idxs)):
 			example_idx, motif_idx, strand_idx, pos_idx = idxs
-			score = scores[*idxs].item()
+			score = scores[(example_idx, motif_idx, strand_idx, pos_idx)].item()
 
 			l = self.motif_lengths[motif_idx]
 			start = pos_idx.item()