Explorar o código

fix(ocr_mkcontent): correct drop mode handling for pages with drop reasons

myhloli hai 1 ano
pai
achega
196de029a3
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      magic_pdf/dict2md/ocr_mkcontent.py

+ 1 - 1
magic_pdf/dict2md/ocr_mkcontent.py

@@ -403,7 +403,7 @@ def union_make(pdf_info_dict: list,
         drop_reason = page_info.get('drop_reason', None)
         if page_info.get('need_drop', False):
             # drop_reason = page_info.get('drop_reason')
-            if drop_mode == DropMode.NONE:
+            if drop_mode == DropMode.NONE or drop_mode == DropMode.NONE_WITH_REASON:
                 pass
             elif drop_mode == DropMode.WHOLE_PDF:
                 raise Exception((f'drop_mode is {DropMode.WHOLE_PDF} ,'