fz0x1/HPI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

allow passing pdfs for testing

Browse source
This commit is contained in:
Dima Gerasimov 2019-09-17 17:42:21 +04:00
parent a016e494c2
commit 2838c8caf0
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
reading/pdfs.py
View file

@ -116,11 +116,12 @@ def test2():
print(res)
def get_annotated_pdfs() -> List[Pdf]:
def get_annotated_pdfs(pdfs=None) -> List[Pdf]:
logger = get_logger()
setup_logzero(logger, level=logging.DEBUG)
pdfs = get_candidates()
if pdfs is None:
pdfs = get_candidates()
logger.info('processing %d pdfs', len(pdfs))
collected = []