Browse Source

remove uttest

master^2
shuaigezhu 3 years ago
parent
commit
65adde14d8
1 changed files with 0 additions and 31 deletions
  1. +0
    -31
      tests/pipelines/test_CodeGeeX_code_translation.py

+ 0
- 31
tests/pipelines/test_CodeGeeX_code_translation.py View File

@@ -1,31 +0,0 @@
# Copyright (c) Alibaba, Inc. and its affiliates.
import os
import unittest

from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.utils.demo_utils import DemoCompatibilityCheck
from modelscope.utils.test_utils import test_level


class CodeGeeXCodeTranslationTest(unittest.TestCase, DemoCompatibilityCheck):

def setUp(self) -> None:
self.output_dir = 'unittest_output'
os.makedirs(self.output_dir, exist_ok=True)

@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
def test_run_with_CodeGeeX_with_name(self):
model = 'ZhipuAI/CodeGeeX-Code-Translation-13B'
pipe = pipeline(task=Tasks.code_translation, model=model)
inputs = {
'prompt': 'for i in range(10):\n\tprint(i)\n',
'source language': 'Python',
'target language': 'C++'
}
result = pipe(inputs)
print(result)


if __name__ == '__main__':
unittest.main()

Loading…
Cancel
Save