Skip to content

Commit 376a2a1

Browse files
authored
fix docify (#2526)
1 parent b2629b6 commit 376a2a1

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

onmt/transforms/docify.py

-4
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,6 @@ def batch_apply(self, batch, is_train=False, stats=None, **kwargs):
6868
doc = {}
6969
doc["src"] = []
7070
doc["tgt"] = []
71-
doc["ind_in_bucket"] = 0
7271
doc["cid"] = ""
7372
doc["cid_line_number"] = 0
7473

@@ -82,7 +81,6 @@ def batch_apply(self, batch, is_train=False, stats=None, **kwargs):
8281
doc = {}
8382
doc["src"] = []
8483
doc["tgt"] = []
85-
doc["ind_in_bucket"] = ex["ind_in_bucket"]
8684
doc["cid"] = ex["cid"]
8785
doc["cid_line_number"] = ex["cid_line_number"]
8886
elif cur_len > self.doc_length:
@@ -110,7 +108,6 @@ def batch_apply(self, batch, is_train=False, stats=None, **kwargs):
110108
doc = {}
111109
doc["src"] = []
112110
doc["tgt"] = []
113-
doc["ind_in_bucket"] = ex["ind_in_bucket"]
114111
doc["cid"] = ex["cid"]
115112
doc["cid_line_number"] = ex["cid_line_number"]
116113
else:
@@ -138,7 +135,6 @@ def batch_apply(self, batch, is_train=False, stats=None, **kwargs):
138135
trf_batch.append((doc, self, cid))
139136
doc = {}
140137
doc["src"] = []
141-
doc["ind_in_bucket"] = ex["ind_in_bucket"]
142138
doc["cid"] = ex["cid"]
143139
doc["cid_line_number"] = ex["cid_line_number"]
144140
if len(doc["src"]) > 0:

0 commit comments

Comments
 (0)