evalscope 1.0.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

tests/swift/test_run_swift_vlm_jugde_eval.py

@@ -1,157 +0,0 @@
-# Copyright (c) Alibaba, Inc. and its affiliates.
-import json
-import os
-import requests
-import shutil
-import subprocess
-import time
-import unittest
-
-from evalscope.backend.vlm_eval_kit import VLMEvalKitBackendManager
-from evalscope.run import run_task
-from evalscope.summarizer import Summarizer
-from evalscope.utils.import_utils import is_module_installed
-from evalscope.utils.logger import get_logger
-from tests.utils import test_level_list
-
-logger = get_logger(__name__)
-
-DEFAULT_CHAT_MODEL_URL = 'http://127.0.0.1:8000/v1/chat/completions'
-DEFAULT_JUDGE_MODEL_URL = 'http://127.0.0.1:8866/v1/chat/completions'
-
-DEFAULT_API_KEY = 'EMPTY'
-DEFAULT_MODEL_NAME = 'CustomAPIModel'
-DEFAULT_WORK_DIR = 'outputs'
-
-
-class TestRunSwiftVLMEval(unittest.TestCase):
-
-    def setUp(self) -> None:
-        logger.info('Init env for swift-eval UTs ...\n')
-        assert is_module_installed('llmuses'), 'Please install `llmuses` from pypi or source code.'
-
-        logger.warning('Note: installing ms-vlmeval ...')
-        subprocess.run('pip3 install ms-vlmeval -U', shell=True, check=True)
-
-        logger.warning('Note: installing ms-swift ...')
-        subprocess.run('pip3 install ms-swift -U', shell=True, check=True)
-
-        if os.path.exists(DEFAULT_WORK_DIR):
-            shutil.rmtree(DEFAULT_WORK_DIR)
-            logger.info(f'Removed work dir: {os.path.abspath(DEFAULT_WORK_DIR)} \n')
-
-        logger.info('\nStaring run swift deploy ...')
-        self.model_name = 'qwen-vl-chat'
-        self.process_swift_deploy = subprocess.Popen(
-            f'swift deploy --model_type {self.model_name}',
-            text=True,
-            shell=True,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE)
-
-        logger.info('\nStaring run swift deploy judge ...')
-        self.judge_model_name = 'qwen2-7b-instruct'
-        self.process_swift_deploy_judge = subprocess.Popen(
-            f'swift deploy --model_type {self.judge_model_name} \
-                                                    --port 8866',
-            text=True,
-            shell=True,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE)
-
-        self.all_datasets = VLMEvalKitBackendManager.list_supported_datasets()
-        assert len(self.all_datasets) > 0, f'Failed to list datasets from VLMEvalKit backend: {self.all_datasets}'
-
-    def tearDown(self) -> None:
-        # Stop the swift deploy model service
-        logger.warning('\nStopping swift deploy ...')
-        self.process_swift_deploy.terminate()
-        self.process_swift_deploy.wait()
-
-        self.process_swift_deploy_judge.terminate()
-        self.process_swift_deploy_judge.wait()
-
-        logger.info('Process swift-deploy terminated successfully.')
-
-    @staticmethod
-    def _check_env(module_name: str):
-        if is_module_installed(module_name):
-            logger.info(f'{module_name} is installed.')
-        else:
-            raise ModuleNotFoundError(f'run: pip install {module_name}')
-
-    @staticmethod
-    def check_service_status(url: str, data: dict, retries: int = 20, delay: int = 10):
-        for i in range(retries):
-            try:
-                logger.info(f'Attempt {i + 1}: Checking service at {url} ...')
-                response = requests.post(
-                    url, data=json.dumps(data), headers={'Content-Type': 'application/json'}, timeout=30)
-                if response.status_code == 200:
-                    logger.info(f'Service at {url} is available !\n\n')
-                    return True
-                else:
-                    logger.info(f'Service at {url} returned status code {response.status_code}.')
-            except requests.exceptions.RequestException as e:
-                logger.info(f'Attempt {i + 1}: An error occurred: {e}')
-
-            time.sleep(delay)
-
-        logger.info(f'Service at {url} is not available after {retries} retries.')
-        return False
-
-    @unittest.skipUnless(0 in test_level_list(), 'skip test in current test level')
-    def test_run_api_with_judge(self):
-        # Check the judge service status
-        data = {'model': self.judge_model_name, 'messages': [{'role': 'user', 'content': 'who are you?'}]}
-        assert self.check_service_status(DEFAULT_JUDGE_MODEL_URL, data=data), \
-            f'Failed to check judge service status: {DEFAULT_JUDGE_MODEL_URL}'
-
-        api_base = DEFAULT_CHAT_MODEL_URL
-        task_cfg = {
-            'eval_backend': 'VLMEvalKit',
-            'eval_config': {
-                'LOCAL_LLM':
-                self.judge_model_name,  # judge model id
-                'OPENAI_API_BASE':
-                DEFAULT_JUDGE_MODEL_URL,  # judge model api
-                'OPENAI_API_KEY':
-                DEFAULT_API_KEY,
-                'data': ['SEEDBench_IMG', 'ChartQA_TEST'],
-                'limit':
-                20,
-                'mode':
-                'all',
-                'model': [{
-                    'api_base': api_base,  # swfit deploy model api
-                    'key': DEFAULT_API_KEY,
-                    'name': DEFAULT_MODEL_NAME,  # must be CustomAPIModel for swift
-                    'temperature': 0.0,
-                    'type': self.model_name
-                }],  # swift model type
-                'nproc':
-                1,
-                'reuse':
-                True,
-                'work_dir':
-                DEFAULT_WORK_DIR
-            }
-        }
-
-        # Check the service status
-        data = {'model': self.model_name, 'messages': [{'role': 'user', 'content': 'who are you?'}]}
-        assert self.check_service_status(api_base, data=data), f'Failed to check service status: {api_base}'
-
-        logger.info(f'>> Start to run task: {task_cfg}')
-
-        run_task(task_cfg)
-
-        logger.info('>> Start to get the report with summarizer ...')
-        report_list = Summarizer.get_report_from_cfg(task_cfg)
-        logger.info(f'\n>> The report list: {report_list}')
-
-        assert len(report_list) > 0, f'Failed to get report list: {report_list}'
-
-
-if __name__ == '__main__':
-    unittest.main(buffer=False)

tests/test_run_all.py

@@ -1,12 +0,0 @@
-# Copyright (c) Alibaba, Inc. and its affiliates.
-
-import subprocess
-
-if __name__ == '__main__':
-    cmd = f'TEST_LEVEL_LIST=0,1 python3 -m unittest discover tests'
-    run_res = subprocess.run(cmd, text=True, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-
-    if run_res.returncode == 0:
-        print(f'>>test_run_all stdout: {run_res.stdout}')
-    else:
-        print(f'>>test_run_all stderr: {run_res.stderr}')

tests/utils.py

@@ -1,13 +0,0 @@
-import os
-
-TEST_LEVEL_LIST = [0, 1]
-# Example: export TEST_LEVEL_LIST=0,1
-TEST_LEVEL_LIST_STR = 'TEST_LEVEL_LIST'
-
-
-def test_level_list():
-    global TEST_LEVEL_LIST
-    if TEST_LEVEL_LIST_STR in os.environ:
-        TEST_LEVEL_LIST = [int(x) for x in os.environ[TEST_LEVEL_LIST_STR].split(',')]
-
-    return TEST_LEVEL_LIST

tests/vlm/__init__.py

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

tests/vlm/test_vlmeval.py

@@ -1,102 +0,0 @@
-# Copyright (c) Alibaba, Inc. and its affiliates.
-from dotenv import dotenv_values
-
-from tests.utils import test_level_list
-
-env = dotenv_values('.env')
-import unittest
-
-from evalscope.run import run_task
-from evalscope.summarizer import Summarizer
-from evalscope.utils.import_utils import is_module_installed
-from evalscope.utils.logger import get_logger
-
-logger = get_logger()
-
-
-class TestVLMEval(unittest.TestCase):
-
-    def setUp(self) -> None:
-        self._check_env('vlmeval')
-
-    def tearDown(self) -> None:
-        pass
-
-    @staticmethod
-    def _check_env(module_name: str):
-        if is_module_installed(module_name):
-            logger.info(f'{module_name} is installed.')
-        else:
-            raise ModuleNotFoundError(f'run: pip install {module_name}')
-
-    @unittest.skipUnless(0 in test_level_list(), 'skip test in current test level')
-    def test_run_vlm_eval_local(self):
-        task_cfg = {
-            'eval_backend': 'VLMEvalKit',
-            'eval_config': {
-                'data': ['SEEDBench_IMG', 'ChartQA_TEST'],
-                'limit': 20,
-                'mode': 'all',
-                'model': [{
-                    'name': 'qwen-vl-chat',
-                    'model_path': '../models/Qwen-VL-Chat'
-                }],  # model name for VLMEval config
-                'nproc': 1,
-                'reuse': True,
-            },
-            'work_dir': 'outputs',
-            'use_cache': 'outputs/20241216_142838'
-        }
-
-        logger.info(f'>> Start to run task: {task_cfg}')
-
-        run_task(task_cfg)
-
-        logger.info('>> Start to get the report with summarizer ...')
-        report_list = Summarizer.get_report_from_cfg(task_cfg)
-        logger.info(f'\n>>The report list: {report_list}')
-
-        assert len(report_list) > 0, f'Failed to get report list: {report_list}'
-
-
-    @unittest.skipUnless(0 in test_level_list(), 'skip test in current test level')
-    def test_run_vlm_api(self):
-        task_cfg = {
-            'eval_backend': 'VLMEvalKit',
-            'eval_config': {
-                'data': [
-                    # 'SEEDBench_IMG',
-                    # 'ChartQA_TEST',
-                    'MMDU'
-                    ],
-                'limit': 5,
-                'mode': 'all',
-                'model': [
-                    {'api_base': 'https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions',
-                    'key': env.get('DASHSCOPE_API_KEY'),
-                    'name': 'CustomAPIModel',
-                    'temperature': 0.0,
-                    'type': 'qwen2.5-vl-7b-instruct',
-                    'img_size': -1,
-                    'video_llm': False,
-                    'max_tokens': 512,}
-                    ],
-                'nproc': 5,
-                'reuse': False,
-            },
-            'work_dir': 'outputs',
-            # 'use_cache': 'outputs/20241216_142838'
-        }
-
-        logger.info(f'>> Start to run task: {task_cfg}')
-
-        run_task(task_cfg)
-
-        logger.info('>> Start to get the report with summarizer ...')
-        report_list = Summarizer.get_report_from_cfg(task_cfg)
-        logger.info(f'\n>>The report list: {report_list}')
-
-        assert len(report_list) > 0, f'Failed to get report list: {report_list}'
-
-if __name__ == '__main__':
-    unittest.main(buffer=False)