From 80c04ca73228d435855f23dbd5b048cda7c5a972 Mon Sep 17 00:00:00 2001 From: liukaiwen Date: Mon, 22 Apr 2024 14:33:23 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E4=BA=86magic=5Fmodel?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- magic_pdf/model/magic_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/magic_pdf/model/magic_model.py b/magic_pdf/model/magic_model.py index bfad7f0c..033dc4a7 100644 --- a/magic_pdf/model/magic_model.py +++ b/magic_pdf/model/magic_model.py @@ -419,7 +419,7 @@ class MagicModel: page_h = page.rect.height return page_w, page_h - def __get_blocks_by_type(self, types: list, page_no: int, extra_col: list[str] = []) -> list: + def __get_blocks_by_type(self, type: int, page_no: int, extra_col: list[str] = []) -> list: blocks = [] for page_dict in self.__model_list: layout_dets = page_dict.get("layout_dets", []) @@ -431,7 +431,7 @@ class MagicModel: category_id = item.get("category_id", -1) bbox = item.get("bbox", None) - if category_id in types: + if category_id == type: block = { "bbox": bbox }