Răsfoiți Sursa

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

myhloli 1 an în urmă
părinte
comite
196de029a3
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  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} ,'