fix parameters mismatch

kermitt2 · Jan 24, 2025 · b3c6049 · b3c6049
1 parent 481ceff
commit b3c6049
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/grobid_client/grobid_client.py b/grobid_client/grobid_client.py
@@ -221,7 +221,9 @@ def process_batch(
                     include_raw_affiliations,
                     tei_coordinates,
                     segment_sentences,
-                    flavor)
+                    flavor,
+                    -1,
+                    -1)
 
                 results.append(r)
 
@@ -261,9 +263,9 @@ def process_pdf(
         include_raw_affiliations,
         tei_coordinates,
         segment_sentences,
+        flavor=None,
         start=-1,
-        end=-1,
-        flavor=None
+        end=-1
     ):
         pdf_handle = open(pdf_file, "rb")
         files = {
@@ -300,6 +302,8 @@ def process_pdf(
         if end > 0:
             the_data["end"] = str(end)
 
+        the_data["includeDiscardedText"] = "1"
+
         try:
             res, status = self.post(
                 url=the_url, files=files, data=the_data, headers={"Accept": "text/plain"}, timeout=self.config['timeout']
@@ -318,7 +322,8 @@ def process_pdf(
                     tei_coordinates,
                     segment_sentences,
                     start,
-                    end
+                    end,
+                    flavor
                 )
         except requests.exceptions.ReadTimeout:
             pdf_handle.close()