Browse Source

fix(magic_pdf): correct negative indexing for `end_page_id`

- Update the logic for determining `end_page_id` to handle negative values
- This change ensures proper behavior when `end_page_id` is set to -1 or other negative values
myhloli 9 months ago
parent
commit
90a27ecd26
1 changed files with 5 additions and 1 deletions
  1. 5 1
      magic_pdf/model/doc_analyze_by_custom_model.py

+ 5 - 1
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -145,7 +145,11 @@ def doc_analyze(
     table_enable=None,
 ) -> InferenceResult:
 
-    end_page_id = end_page_id if end_page_id is not None else len(dataset) - 1
+    end_page_id = (
+        end_page_id
+        if end_page_id is not None and end_page_id >= 0
+        else len(dataset) - 1
+    )
 
     model_manager = ModelSingleton()
     custom_model = model_manager.get_model(