mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-24 12:09:22 +01:00
remove uttest
This commit is contained in:
@@ -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()
|
||||
Reference in New Issue
Block a user