ci: fix unit testing error

RapidAI · Sep 12, 2024 · ad4a3ed · ad4a3ed
1 parent 4d03958
commit ad4a3ed
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 13 deletions.
diff --git a/.github/workflows/lineless_table_rec.yml b/.github/workflows/lineless_table_rec.yml
@@ -15,10 +15,10 @@ jobs:
       - name: Pull latest code
         uses: actions/checkout@v3
 
-      - name: Set up Python 3.7
+      - name: Set up Python 3.10
         uses: actions/setup-python@v4
         with:
-          python-version: '3.7'
+          python-version: '3.10'
           architecture: 'x64'
 
       - name: Display Python version
@@ -33,7 +33,7 @@ jobs:
           unzip lineless_table_rec_models.zip
           mv lineless_table_rec_models/*.onnx lineless_table_rec/models/
 
-          pytest tests/test_lore.py
+          pytest tests/test_lineless_table_rec.py
 
   GenerateWHL_PushPyPi:
     needs: UnitTesting

diff --git a/.github/workflows/wired_table_rec.yml b/.github/workflows/wired_table_rec.yml
@@ -15,10 +15,10 @@ jobs:
       - name: Pull latest code
         uses: actions/checkout@v3
 
-      - name: Set up Python 3.7
+      - name: Set up Python 3.10
         uses: actions/setup-python@v4
         with:
-          python-version: '3.7'
+          python-version: '3.10'
           architecture: 'x64'
 
       - name: Display Python version
@@ -33,7 +33,7 @@ jobs:
           unzip wired_table_rec_models.zip
           mv wired_table_rec_models/*.onnx wired_table_rec/models/
 
-          pytest tests/test_wired.py
+          pytest tests/test_wired_table_rec.py
 
   GenerateWHL_PushPyPi:
     needs: UnitTesting
@@ -60,13 +60,6 @@ jobs:
 
           python setup_wired.py bdist_wheel "${{ github.event.head_commit.message }}"
 
-      # - name: Publish distribution 📦 to Test PyPI
-      #   uses: pypa/[email protected]
-      #   with:
-      #     password: ${{ secrets.TEST_PYPI_API_TOKEN }}
-      #     repository_url: https://test.pypi.org/legacy/
-      #     packages_dir:  dist/
-
       - name: Publish distribution 📦 to PyPI
         uses: pypa/[email protected]
         with:

diff --git a/lineless_table_rec/main.py b/lineless_table_rec/main.py
@@ -199,6 +199,7 @@ def handle_overlap_row_col(self, res):
 
         # 创建一个二维数组来存储 sorted_logi_points 中的元素
         grid = [[None] * max_col for _ in range(max_row)]
+
         # 将 sorted_logi_points 中的元素填充到 grid 中
         deleted_idx = set()
         for i, dict_res in enumerate(res):