"tests/vscode:/vscode.git/clone" did not exist on "80f27d7e8db9a6d9a79a320171446963660d8cdf"
Commit c064379c authored by myhloli's avatar myhloli
Browse files

refactor(test_table): update table model class name

- Rename ppTableModel to TableMasterPaddleModel in test_tablemaster.py
parent 08f46125
......@@ -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()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment