centersl/MinerU
1
0
Fork 0
mirror of https://github.com/opendatalab/MinerU.git synced 2026-03-27 11:08:32 +07:00
Code Issues Packages Projects Releases Wiki Activity

refactor(test_table): update table model class name

Browse Source 
- Rename ppTableModel to TableMasterPaddleModel in test_tablemaster.py
This commit is contained in:
myhloli
2024-11-15 18:56:08 +08:00
parent 08f46125a0
commit c064379cd1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_table/test_tablemaster.py
View File

@@ -2,7 +2,7 @@ import unittest
from PIL import Image
from lxml import etree
from magic_pdf.model.sub_modules.table.tablemaster.tablemaster_paddle import ppTableModel
from magic_pdf.model.sub_modules.table.tablemaster.tablemaster_paddle import TableMasterPaddleModel
class TestppTableModel(unittest.TestCase):
@@ -11,7 +11,7 @@ class TestppTableModel(unittest.TestCase):
# 修改table模型路径
config = {"device": "cuda",
"model_dir": "/home/quyuan/.cache/modelscope/hub/opendatalab/PDF-Extract-Kit/models/TabRec/TableMaster"}
table_model = ppTableModel(config)
table_model = TableMasterPaddleModel(config)
res = table_model.img2html(img)
# 验证生成的 HTML 是否符合预期
parser = etree.HTMLParser()