diff --git a/gnes/preprocessor/text/split.py b/gnes/preprocessor/text/split.py index 5aa2b7a6..5e89838e 100644 --- a/gnes/preprocessor/text/split.py +++ b/gnes/preprocessor/text/split.py @@ -53,6 +53,6 @@ def apply(self, doc: 'gnes_pb2.Document') -> None: c = doc.chunks.add() c.doc_id = doc.doc_id c.text = f[:self.max_sent_len] - c.offset_1d = ci + c.offset = ci c.weight = len(c.text) / len(doc.raw_text) c.offset_nd.extend([s, e])