Bläddra i källkod

refactor(test_table): update table model class name

- Rename ppTableModel to TableMasterPaddleModel in test_tablemaster.py
myhloli 1 år sedan
förälder
incheckning
c064379cd1
1 ändrade filer med 2 tillägg och 2 borttagningar
  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 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()