diff --git a/tangermeme/tools/fimo.py b/tangermeme/tools/fimo.py index d26e264..08a310b 100644 --- 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()