mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
13 lines
352 B
Python
13 lines
352 B
Python
|
from mmengine.utils import get_git_hash
|
||
|
|
from mmengine.utils.dl_utils import collect_env as collect_base_env
|
||
|
|
|
||
|
|
import opencompass
|
||
|
|
|
||
|
|
|
||
|
|
def collect_env():
|
||
|
|
"""Collect the information of the running environments."""
|
||
|
|
env_info = collect_base_env()
|
||
|
|
env_info['opencompass'] = opencompass.__version__ + '+' + get_git_hash(
|
||
|
|
)[:7]
|
||
|
|
return env_info
|