Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-24 03:59:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

add 1.6

Browse Source
This commit is contained in:
xingjun.wang
2023-05-22 10:53:18 +08:00
parent 52aea36c12
commit 48c0d2a9af
468 changed files with 12942 additions and 7176 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/pipelines/test_document_segmentation.py
View File

@@ -6,14 +6,13 @@ from typing import Any, Dict
from modelscope.outputs import OutputKeys
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.utils.demo_utils import DemoCompatibilityCheck
from modelscope.utils.logger import get_logger
from modelscope.utils.test_utils import test_level
logger = get_logger()
class DocumentSegmentationTest(unittest.TestCase, DemoCompatibilityCheck):
class DocumentSegmentationTest(unittest.TestCase):
def setUp(self) -> None:
self.task = Tasks.document_segmentation
@@ -64,10 +63,6 @@ class DocumentSegmentationTest(unittest.TestCase, DemoCompatibilityCheck):
for document in documents_list:
print(document)
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_demo_compatibility(self):
self.compatibility_check()
if __name__ == '__main__':
unittest.main()