diff options
-rw-r--r-- | python/sandcrawler/grobid.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/python/sandcrawler/grobid.py b/python/sandcrawler/grobid.py index a44861a..ae96fc8 100644 --- a/python/sandcrawler/grobid.py +++ b/python/sandcrawler/grobid.py @@ -28,13 +28,14 @@ class GrobidClient(object): if consolidate_mode is None: consolidate_mode = self.consolidate_mode + assert consolidate_mode is not None try: grobid_response = requests.post( self.host_url + "/api/processFulltextDocument", files={ 'input': blob, - 'consolidateHeader': self.consolidate_mode, + 'consolidateHeader': consolidate_mode, 'consolidateCitations': 0, # too expensive for now 'includeRawCitations': 1, }, |