Browse Source

revise modelhub dependency

master
智丞 3 years ago
parent
commit
c63107c500
6 changed files with 6 additions and 12 deletions
  1. +1
    -2
      tests/pipelines/nlp/test_dialog_intent_prediction.py
  2. +1
    -2
      tests/pipelines/nlp/test_dialog_modeling.py
  3. +1
    -2
      tests/pipelines/test_fill_mask.py
  4. +1
    -2
      tests/pipelines/test_nli.py
  5. +1
    -2
      tests/pipelines/test_sentiment_classification.py
  6. +1
    -2
      tests/pipelines/test_zero_shot_classification.py

+ 1
- 2
tests/pipelines/nlp/test_dialog_intent_prediction.py View File

@@ -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


+ 1
- 2
tests/pipelines/nlp/test_dialog_modeling.py View File

@@ -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


+ 1
- 2
tests/pipelines/test_fill_mask.py View File

@@ -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


+ 1
- 2
tests/pipelines/test_nli.py View File

@@ -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


+ 1
- 2
tests/pipelines/test_sentiment_classification.py View File

@@ -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


+ 1
- 2
tests/pipelines/test_zero_shot_classification.py View File

@@ -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


Loading…
Cancel
Save