mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-24 12:09:22 +01:00
5ef842e38a
* allow token classification pipelines to predict longer sentences * bugfix * skip adaseq pipeline ut when connection error occurs * return entity probabilities
54 lines
2.2 KiB
Python
54 lines
2.2 KiB
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
||
import unittest
|
||
|
||
from modelscope.pipelines import pipeline
|
||
from modelscope.utils.constant import Tasks
|
||
from modelscope.utils.plugins import PluginsManager
|
||
from modelscope.utils.test_utils import test_level
|
||
|
||
|
||
class PluginModelTest(unittest.TestCase):
|
||
|
||
def setUp(self):
|
||
self.package = 'adaseq'
|
||
|
||
def tearDown(self):
|
||
# make sure uninstalled after installing
|
||
super().tearDown()
|
||
import subprocess
|
||
result = subprocess.run(
|
||
['pip', 'install', 'adaseq>=0.6.2', '--no-deps'],
|
||
stdout=subprocess.PIPE)
|
||
print(result.stdout.decode('utf-8'))
|
||
|
||
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
|
||
def test_run_span_based_ner_pipeline(self):
|
||
try:
|
||
pipeline_ins = pipeline(
|
||
Tasks.named_entity_recognition,
|
||
'damo/nlp_nested-ner_named-entity-recognition_chinese-base-med'
|
||
)
|
||
print(
|
||
pipeline_ins(
|
||
'1、可测量目标： 1周内胸闷缓解。2、下一步诊疗措施：1.心内科护理常规，一级护理，低盐低脂饮食，留陪客。'
|
||
'2.予“阿司匹林肠溶片”抗血小板聚集，“呋塞米、螺内酯”利尿减轻心前负荷，“瑞舒伐他汀”调脂稳定斑块，“厄贝沙坦片片”降血压抗心机重构'
|
||
))
|
||
except RuntimeError:
|
||
print(
|
||
'Skip test span_based_ner_pipeline! RuntimeError: Try loading from huggingface and modelscope failed'
|
||
)
|
||
|
||
def test_maoe_pipelines(self):
|
||
try:
|
||
pipeline_ins = pipeline(
|
||
Tasks.named_entity_recognition,
|
||
'damo/nlp_maoe_named-entity-recognition_chinese-base-general')
|
||
print(
|
||
pipeline_ins(
|
||
'刘培强，男，生理年龄40岁（因为在太空中进入休眠状态），实际年龄52岁，领航员国际空间站中的中国航天员，机械工程专家，军人，军衔中校。'
|
||
))
|
||
except RuntimeError:
|
||
print(
|
||
'Skip test maoe_pipeline! RuntimeError: Try loading from huggingface and modelscope failed'
|
||
)
|