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