diff --git a/tests/pipelines/nlp/test_dialog_intent_prediction.py b/tests/pipelines/nlp/test_dialog_intent_prediction.py index 0ec4e1e7..97cdbb3d 100644 --- a/tests/pipelines/nlp/test_dialog_intent_prediction.py +++ b/tests/pipelines/nlp/test_dialog_intent_prediction.py @@ -1,8 +1,7 @@ # Copyright (c) Alibaba, Inc. and its affiliates. import unittest -from maas_hub.snapshot_download import snapshot_download - +from modelscope.hub.snapshot_download import snapshot_download from modelscope.models import Model from modelscope.models.nlp import DialogIntentModel from modelscope.pipelines import DialogIntentPredictionPipeline, pipeline diff --git a/tests/pipelines/nlp/test_dialog_modeling.py b/tests/pipelines/nlp/test_dialog_modeling.py index 7d4da8fe..f606ba49 100644 --- a/tests/pipelines/nlp/test_dialog_modeling.py +++ b/tests/pipelines/nlp/test_dialog_modeling.py @@ -4,8 +4,7 @@ import os.path as osp import tempfile import unittest -from maas_hub.snapshot_download import snapshot_download - +from modelscope.hub.snapshot_download import snapshot_download from modelscope.models import Model from modelscope.models.nlp import DialogModelingModel from modelscope.pipelines import DialogModelingPipeline, pipeline diff --git a/tests/pipelines/test_fill_mask.py b/tests/pipelines/test_fill_mask.py index a4d53403..886fa70f 100644 --- a/tests/pipelines/test_fill_mask.py +++ b/tests/pipelines/test_fill_mask.py @@ -3,8 +3,7 @@ import os import shutil import unittest -from maas_hub.snapshot_download import snapshot_download - +from modelscope.hub.snapshot_download import snapshot_download from modelscope.models import Model from modelscope.models.nlp import StructBertForMaskedLM, VecoForMaskedLM from modelscope.pipelines import FillMaskPipeline, pipeline diff --git a/tests/pipelines/test_nli.py b/tests/pipelines/test_nli.py index ad94697a..0fb0bd2b 100644 --- a/tests/pipelines/test_nli.py +++ b/tests/pipelines/test_nli.py @@ -1,8 +1,7 @@ # Copyright (c) Alibaba, Inc. and its affiliates. import unittest -from maas_hub.snapshot_download import snapshot_download - +from modelscope.hub.snapshot_download import snapshot_download from modelscope.models import Model from modelscope.models.nlp import SbertForNLI from modelscope.pipelines import NLIPipeline, pipeline diff --git a/tests/pipelines/test_sentiment_classification.py b/tests/pipelines/test_sentiment_classification.py index 9a1a8484..7e55cd4c 100644 --- a/tests/pipelines/test_sentiment_classification.py +++ b/tests/pipelines/test_sentiment_classification.py @@ -1,8 +1,7 @@ # Copyright (c) Alibaba, Inc. and its affiliates. import unittest -from maas_hub.snapshot_download import snapshot_download - +from modelscope.hub.snapshot_download import snapshot_download from modelscope.models import Model from modelscope.models.nlp import SbertForSentimentClassification from modelscope.pipelines import SentimentClassificationPipeline, pipeline diff --git a/tests/pipelines/test_zero_shot_classification.py b/tests/pipelines/test_zero_shot_classification.py index 2c32142a..63229dfe 100644 --- a/tests/pipelines/test_zero_shot_classification.py +++ b/tests/pipelines/test_zero_shot_classification.py @@ -1,8 +1,7 @@ # Copyright (c) Alibaba, Inc. and its affiliates. import unittest -from maas_hub.snapshot_download import snapshot_download - +from modelscope.hub.snapshot_download import snapshot_download from modelscope.models import Model from modelscope.models.nlp import BertForZeroShotClassification from modelscope.pipelines import ZeroShotClassificationPipeline, pipeline