From f3ceebc41bc728f6113dc39d2c84808667b7143a Mon Sep 17 00:00:00 2001 From: xu rui Date: Tue, 10 Dec 2024 10:54:44 +0800 Subject: [PATCH] fix: read_api list files --- magic_pdf/data/read_api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/magic_pdf/data/read_api.py b/magic_pdf/data/read_api.py index 58089421..507b7f11 100644 --- a/magic_pdf/data/read_api.py +++ b/magic_pdf/data/read_api.py @@ -58,7 +58,7 @@ def read_local_pdfs(path: str) -> list[PymuDocDataset]: list[PymuDocDataset]: each pdf file will converted to a PymuDocDataset """ if os.path.isdir(path): - reader = FileBasedDataReader(path) + reader = FileBasedDataReader() ret = [] for root, _, files in os.walk(path): for file in files: @@ -85,7 +85,7 @@ def read_local_images(path: str, suffixes: list[str]=[]) -> list[ImageDataset]: if os.path.isdir(path): imgs_bits = [] s_suffixes = set(suffixes) - reader = FileBasedDataReader(path) + reader = FileBasedDataReader() for root, _, files in os.walk(path): for file in files: suffix = file.split('.')