From c241b97fae10ff8eb26fcbe83a3813e523b1423a Mon Sep 17 00:00:00 2001 From: Louis Chartrand Date: Tue, 16 Mar 2021 15:09:34 -0400 Subject: [PATCH] bugfix --- clctm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clctm.py b/clctm.py index 102c02c..e028103 100644 --- a/clctm.py +++ b/clctm.py @@ -131,7 +131,7 @@ def _get_indices(self, doc_idx): if isinstance(doc_idx, int): return list(range(*self.doc_rng(doc_idx))) else: - return list(chain(*(range(*self.doc_rng[d] for d in doc_idx)))) + return list(chain(*(range(*self.doc_rng[d]) for d in doc_idx))) def _get_mask(self, doc_idx): r = np.full(len(self.input_ids), False)