diff --git a/grobid_client/grobid_client.py b/grobid_client/grobid_client.py index 3d585bf..56d4ff6 100644 --- a/grobid_client/grobid_client.py +++ b/grobid_client/grobid_client.py @@ -720,7 +720,10 @@ def process_txt( include_raw_citations, include_raw_affiliations, tei_coordinates, - segment_sentences + segment_sentences, + flavor=None, + start_page=-1, + end_page=-1 ): # create request based on file content try: diff --git a/tests/test_grobid_client.py b/tests/test_grobid_client.py index 98adb42..b6ab324 100644 --- a/tests/test_grobid_client.py +++ b/tests/test_grobid_client.py @@ -1,14 +1,13 @@ """ Unit tests for the GROBID client main functionality. """ -import pytest -from unittest.mock import Mock, patch, MagicMock, mock_open, call import json import os import tempfile -import logging +from unittest.mock import Mock, patch, mock_open + +import pytest import requests -from io import StringIO from grobid_client.grobid_client import GrobidClient, ServerUnavailableException