浏览代码

refactor(test_table): update table model class name

- Rename ppTableModel to TableMasterPaddleModel in test_tablemaster.py
myhloli 1 年之前
父节点
当前提交
c064379cd1
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      tests/test_table/test_tablemaster.py

+ 2 - 2
tests/test_table/test_tablemaster.py

@@ -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):
         # 修改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()