xu rui 11 місяців тому
батько
коміт
f3ceebc41b
1 змінених файлів з 2 додано та 2 видалено
  1. 2 2
      magic_pdf/data/read_api.py

+ 2 - 2
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('.')