Skip to content

Commit 8b1ec95

Browse files
authored
Merge pull request #99 from kermitt2/bugfix/process-citations
Fix processCitationList
2 parents 8b805f5 + d77252e commit 8b1ec95

2 files changed

Lines changed: 7 additions & 5 deletions

File tree

grobid_client/grobid_client.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -720,7 +720,10 @@ def process_txt(
720720
include_raw_citations,
721721
include_raw_affiliations,
722722
tei_coordinates,
723-
segment_sentences
723+
segment_sentences,
724+
flavor=None,
725+
start_page=-1,
726+
end_page=-1
724727
):
725728
# create request based on file content
726729
try:

tests/test_grobid_client.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,13 @@
11
"""
22
Unit tests for the GROBID client main functionality.
33
"""
4-
import pytest
5-
from unittest.mock import Mock, patch, MagicMock, mock_open, call
64
import json
75
import os
86
import tempfile
9-
import logging
7+
from unittest.mock import Mock, patch, mock_open
8+
9+
import pytest
1010
import requests
11-
from io import StringIO
1211

1312
from grobid_client.grobid_client import GrobidClient, ServerUnavailableException
1413

0 commit comments

Comments
 (0)