Skip to content

Commit 9ef0ad0

Browse files
chg: exmplain arbitrary 15
1 parent 55bc5e2 commit 9ef0ad0

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

deep_reference_parser/prodigy/prodigy_to_tsv.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@
2020

2121
msg = Printer()
2222

23+
ROWS_TO_PRINT=15
24+
2325

2426
class TokenLabelPairs:
2527
"""
@@ -365,11 +367,11 @@ def prodigy_to_tsv(
365367
example_pairs = [
366368
combine_token_label_pairs(pairs)
367369
for i, pairs in enumerate(reduce(zip, pairs_list))
368-
if i < 15
370+
if i < ROWS_TO_PRINT
369371
]
370372
else:
371373
merged_pairs = pairs_list[0]
372-
example_pairs = merged_pairs[0:14]
374+
example_pairs = merged_pairs[0:ROWS_TO_PRINT]
373375

374376
with open(output_file, "w") as fb:
375377
writer = csv.writer(fb, delimiter="\t")

0 commit comments

Comments
 (0)