瀏覽代碼

fix task_service.py (#8687)

Fix the case where pages variable might be None

### What problem does this PR solve?


### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.20.0
Fee He 3 月之前
父節點
當前提交
ae3683c346
No account linked to committer's email address
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2
    0
      api/db/services/task_service.py

+ 2
- 0
api/db/services/task_service.py 查看文件

@@ -338,6 +338,8 @@ def queue_tasks(doc: dict, bucket: str, name: str, priority: int):
file_bin = STORAGE_IMPL.get(bucket, name)
do_layout = doc["parser_config"].get("layout_recognize", "DeepDOC")
pages = PdfParser.total_page_number(doc["name"], file_bin)
if pages is None:
pages = 0
page_size = doc["parser_config"].get("task_page_size") or 12
if doc["parser_id"] == "paper":
page_size = doc["parser_config"].get("task_page_size") or 22

Loading…
取消
儲存