Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-02 03:59:31 +02:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933] skip demo test by default

Browse Source
This commit is contained in:
Yingda Chen
2022-09-09 14:56:15 +08:00
parent a4cfbaa0dd
commit e0ef60ca9b
49 changed files with 49 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/pipelines/test_automatic_speech_recognition.py
View File

@@ -257,7 +257,7 @@ class AutomaticSpeechRecognitionTest(unittest.TestCase,
model_id=self.am_tf_model_id, audio_in=dataset_path)
self.check_result('test_run_with_wav_dataset_tf', rec_result)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_cmdssl_video_embedding.py
View File

@@ -22,7 +22,7 @@ class CMDSSLVideoEmbeddingTest(unittest.TestCase, DemoCompatibilityCheck):
print(f'video embedding output: {result}.')
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_generative_multi_modal_embedding.py
View File

@@ -68,7 +68,7 @@ class GEMMMultiModalEmbeddingTest(unittest.TestCase, DemoCompatibilityCheck):
output = generative_multi_modal_embedding_pipeline(test_input)
print(output)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_hicossl_video_embedding.py
View File

@@ -23,7 +23,7 @@ class HICOSSLVideoEmbeddingTest(unittest.TestCase, DemoCompatibilityCheck):
print(f'video embedding output: {result}.')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_colorization.py
View File

@@ -37,7 +37,7 @@ class ImageColorizationTest(unittest.TestCase, DemoCompatibilityCheck):
image_colorization = pipeline(Tasks.image_colorization)
self.pipeline_inference(image_colorization, self.test_image)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_denoise.py
View File

@@ -61,7 +61,7 @@ class ImageDenoiseTest(unittest.TestCase, DemoCompatibilityCheck):
w, h = denoise_img.size
print('pipeline: the shape of output_img is {}x{}'.format(h, w))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_instance_segmentation.py
View File

@@ -61,7 +61,7 @@ class ImageInstanceSegmentationTest(unittest.TestCase, DemoCompatibilityCheck):
print(f'pipeline1:{pipeline1(input=self.image)[OutputKeys.LABELS]}')
print(f'pipeline2: {pipeline2(input=self.image)[OutputKeys.LABELS]}')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_matting.py
View File

@@ -61,7 +61,7 @@ class ImageMattingTest(unittest.TestCase, DemoCompatibilityCheck):
f'Output written to dir: {osp.dirname(osp.abspath("result_0.png"))}'
)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_panoptic_segmentation.py
View File

@@ -38,7 +38,7 @@ class ImagePanopticSegmentationTest(unittest.TestCase, DemoCompatibilityCheck):
cv2.imwrite('result.jpg', draw_img)
print('print test_image_panoptic_segmentation from PIL return success')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_portrait_enhancement.py
View File

@@ -39,7 +39,7 @@ class ImagePortraitEnhancementTest(unittest.TestCase, DemoCompatibilityCheck):
face_enhancement = pipeline(Tasks.image_portrait_enhancement)
self.pipeline_inference(face_enhancement, self.test_image)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_reid_person.py
View File

@@ -50,7 +50,7 @@ class ImageReidPersonTest(unittest.TestCase, DemoCompatibilityCheck):
)
print(f'The img embedding is: {result[OutputKeys.IMG_EMBEDDING]}')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_semantic_segmentation.py
View File

@@ -51,7 +51,7 @@ class ImageSemanticSegmentationTest(unittest.TestCase, DemoCompatibilityCheck):
cv2.imwrite('result.jpg', draw_img)
print('test_image_semantic_segmentation_vitadapter_from_PIL DONE')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_style_transfer.py
View File

@@ -50,7 +50,7 @@ class ImageStyleTransferTest(unittest.TestCase, DemoCompatibilityCheck):
cv2.imwrite('result_styletransfer3.png', result[OutputKeys.OUTPUT_IMG])
print('style_transfer.test_run_modelhub_default_model done')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_image_super_resolution.py
View File

@@ -37,7 +37,7 @@ class ImageSuperResolutionTest(unittest.TestCase, DemoCompatibilityCheck):
super_resolution = pipeline(Tasks.image_super_resolution)
self.pipeline_inference(super_resolution, self.img)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_key_word_spotting.py
View File

@@ -296,7 +296,7 @@ class KeyWordSpottingTest(unittest.TestCase, DemoCompatibilityCheck):
model_id=self.model_id, audio_in=audio_list)
self.check_result('test_run_with_roc', kws_result)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_live_category.py
View File

@@ -21,7 +21,7 @@ class LiveCategoryTest(unittest.TestCase, DemoCompatibilityCheck):
print(f'live category output: {result}.')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_movie_scene_segmentation.py
View File

@@ -35,7 +35,7 @@ class MovieSceneSegmentationTest(unittest.TestCase, DemoCompatibilityCheck):
else:
raise ValueError('process error')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_mplug_tasks.py
View File

@@ -80,7 +80,7 @@ class MplugTasksTest(unittest.TestCase, DemoCompatibilityCheck):
result = pipeline_retrieval(input)
print(result)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_multi_modal_embedding.py
View File

@@ -59,7 +59,7 @@ class MultiModalEmbeddingTest(unittest.TestCase, DemoCompatibilityCheck):
print('l2-norm: {}'.format(torch.norm(text_embedding,
dim=-1).item())) # should be 1.0
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_named_entity_recognition.py
View File

@@ -94,7 +94,7 @@ class NamedEntityRecognitionTest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(task=Tasks.named_entity_recognition)
print(pipeline_ins(input=self.sentence))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_nli.py
View File

@@ -57,7 +57,7 @@ class NLITest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(task=Tasks.nli)
print(pipeline_ins(input=(self.sentence1, self.sentence2)))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_object_detection.py
View File

@@ -55,7 +55,7 @@ class ObjectDetectionTest(unittest.TestCase, DemoCompatibilityCheck):
else:
raise ValueError('process error')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_ocr_detection.py
View File

@@ -30,7 +30,7 @@ class OCRDetectionTest(unittest.TestCase, DemoCompatibilityCheck):
ocr_detection = pipeline(Tasks.ocr_detection)
self.pipeline_inference(ocr_detection, self.test_image)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_ocr_recognition.py
View File

@@ -37,7 +37,7 @@ class OCRRecognitionTest(unittest.TestCase, DemoCompatibilityCheck):
ocr_recognition = pipeline(Tasks.ocr_recognition)
self.pipeline_inference(ocr_recognition, self.test_image)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_ofa_tasks.py
View File

@@ -252,7 +252,7 @@ class OfaTasksTest(unittest.TestCase, DemoCompatibilityCheck):
result[OutputKeys.OUTPUT_IMG].save('result.png')
print(f'Output written to {osp.abspath("result.png")}')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_person_image_cartoon.py
View File

@@ -36,7 +36,7 @@ class ImageCartoonTest(unittest.TestCase, DemoCompatibilityCheck):
img_cartoon = pipeline(Tasks.image_portrait_stylization)
self.pipeline_inference(img_cartoon, self.test_image)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_product_retrieval_embedding.py
View File

@@ -39,7 +39,7 @@ class ProductRetrievalEmbeddingTest(unittest.TestCase, DemoCompatibilityCheck):
result = product_embed(self.img_input)[OutputKeys.IMG_EMBEDDING]
print('abs sum value is: {}'.format(np.sum(np.abs(result))))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_realtime_object_detection.py
View File

@@ -47,7 +47,7 @@ class RealtimeObjectDetectionTest(unittest.TestCase, DemoCompatibilityCheck):
else:
raise ValueError('process error')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_relation_extraction.py
View File

@@ -55,7 +55,7 @@ class RelationExtractionTest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(task=Tasks.information_extraction)
print(pipeline_ins(input=self.sentence))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_salient_detection.py
View File

@@ -24,7 +24,7 @@ class SalientDetectionTest(unittest.TestCase, DemoCompatibilityCheck):
# result[OutputKeys.MASKS] is salient map result,other keys are not used
cv2.imwrite(input_location + '_salient.jpg', result[OutputKeys.MASKS])
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_sentence_similarity.py
View File

@@ -63,7 +63,7 @@ class SentenceSimilarityTest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(task=Tasks.sentence_similarity)
print(pipeline_ins(input=(self.sentence1, self.sentence2)))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_sentiment_classification.py
View File

@@ -66,7 +66,7 @@ class SentimentClassificationTaskModelTest(unittest.TestCase,
self.assertTrue(
isinstance(pipeline_ins.model, SequenceClassificationModel))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_skin_retouching.py
View File

@@ -41,7 +41,7 @@ class SkinRetouchingTest(unittest.TestCase, DemoCompatibilityCheck):
skin_retouching = pipeline(Tasks.skin_retouching)
self.pipeline_inference(skin_retouching, self.test_image)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_speech_signal_process.py
View File

@@ -113,7 +113,7 @@ class SpeechSignalProcessTest(unittest.TestCase, DemoCompatibilityCheck):
ans(data, output_path=output_path)
print(f'Processed audio saved to {output_path}')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_text_classification.py
View File

@@ -89,7 +89,7 @@ class SequenceClassificationTest(unittest.TestCase, DemoCompatibilityCheck):
result = text_classification(dataset)
self.printDataset(result)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_text_driven_segmentation.py
View File

@@ -23,7 +23,7 @@ class TextDrivenSegmentationTest(unittest.TestCase):
# result[OutputKeys.MASKS] is segment map result,other keys are not used
cv2.imwrite(input_location + '_lseg.jpg', result[OutputKeys.MASKS])
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.test_demo()

2
tests/pipelines/test_text_error_correction.py
View File

@@ -55,7 +55,7 @@ class TextErrorCorrectionTest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(task=Tasks.text_error_correction)
print(pipeline_ins(self.input))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_text_generation.py
View File

@@ -129,7 +129,7 @@ class TextGenerationTest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(task=Tasks.text_generation)
print(pipeline_ins(self.palm_input_zh))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_text_to_image_synthesis.py
View File

@@ -51,7 +51,7 @@ class TextToImageSynthesisTest(unittest.TestCase, DemoCompatibilityCheck):
self.test_text)[OutputKeys.OUTPUT_IMG]
print(np.sum(np.abs(img)))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_text_to_speech.py
View File

@@ -38,7 +38,7 @@ class TextToSpeechSambertHifigan16kPipelineTest(unittest.TestCase,
pcm = output[OutputKeys.OUTPUT_PCM]
write('output.wav', 16000, pcm)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_tinynas_classification.py
View File

@@ -19,7 +19,7 @@ class TinyNASClassificationTest(unittest.TestCase, DemoCompatibilityCheck):
result = tinynas_classification('data/test/images/image_wolf.jpeg')
print(result)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_tinynas_detection.py
View File

@@ -15,7 +15,7 @@ class TinynasObjectDetectionTest(unittest.TestCase):
'data/test/images/image_detection.jpg')
print(result)
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.test_demo()

2
tests/pipelines/test_video_category.py
View File

@@ -21,7 +21,7 @@ class VideoCategoryTest(unittest.TestCase, DemoCompatibilityCheck):
print(f'video category output: {result}.')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_video_multi_modal_embedding.py
View File

@@ -41,7 +41,7 @@ class VideoMultiModalEmbeddingTest(unittest.TestCase, DemoCompatibilityCheck):
logger.info('video feature: {}'.format(
output['video_embedding'][0][0][0]))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_video_single_object_tracking.py
View File

@@ -35,7 +35,7 @@ class SingleObjectTracking(unittest.TestCase, DemoCompatibilityCheck):
result = video_single_object_tracking((video_path, init_bbox))
print('result is : ', result[OutputKeys.BOXES])
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_video_summarization.py
View File

@@ -33,7 +33,7 @@ class VideoSummarizationTest(unittest.TestCase, DemoCompatibilityCheck):
print(f'video summarization output:\n {result}.')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_virtual_try_on.py
View File

@@ -34,7 +34,7 @@ class VirtualTryonTest(unittest.TestCase, DemoCompatibilityCheck):
img = pipeline_virtual_tryon(self.input_imgs)[OutputKeys.OUTPUT_IMG]
cv2.imwrite('demo.jpg', img[:, :, ::-1])
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_word_segmentation.py
View File

@@ -59,7 +59,7 @@ class WordSegmentationTest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(task=Tasks.word_segmentation)
print(pipeline_ins(input=self.sentence))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()

2
tests/pipelines/test_zero_shot_classification.py
View File

@@ -70,7 +70,7 @@ class ZeroShotClassificationTest(unittest.TestCase, DemoCompatibilityCheck):
pipeline_ins = pipeline(task=Tasks.zero_shot_classification)
print(pipeline_ins(input=self.sentence, candidate_labels=self.labels))
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@unittest.skip('demo compatibility test is only enabled on a needed-basis')
def test_demo_compatibility(self):
self.compatibility_check()