diff --git a/tsvtools/tsv.py b/tsvtools/tsv.py index 409f475..799b1b5 100644 --- a/tsvtools/tsv.py +++ b/tsvtools/tsv.py @@ -15,7 +15,7 @@ def read_tsv(tsv_file): def write_tsv(tsv, urls, tsv_out_file): - if 'conf' in tsv.columns(): + if 'conf' in tsv.columns: out_columns = ['No.', 'TOKEN', 'NE-TAG', 'NE-EMB', 'ID', 'url_id', 'left', 'right', 'top', 'bottom', 'conf'] else: out_columns = ['No.', 'TOKEN', 'NE-TAG', 'NE-EMB', 'ID', 'url_id', 'left', 'right', 'top', 'bottom']