()
| 105 | |
| 106 | |
| 107 | def do_setup(): |
| 108 | setup( |
| 109 | name='opencompass', |
| 110 | author='OpenCompass Contributors', |
| 111 | version=get_version(), |
| 112 | description='A comprehensive toolkit for large model evaluation', |
| 113 | url='https://github.com/open-compass/opencompass', |
| 114 | long_description=readme(), |
| 115 | long_description_content_type='text/markdown', |
| 116 | maintainer='OpenCompass Authors', |
| 117 | cmdclass={'download_nltk': DownloadNLTK}, |
| 118 | setup_requires=['nltk==3.8'], |
| 119 | python_requires='>=3.8.0', |
| 120 | install_requires=parse_requirements('requirements/runtime.txt'), |
| 121 | extras_require={ |
| 122 | 'lmdeploy': |
| 123 | parse_requirements('requirements/lmdeploy.txt') + |
| 124 | parse_requirements('requirements/runtime.txt'), |
| 125 | 'vllm': |
| 126 | parse_requirements('requirements/vllm.txt') + |
| 127 | parse_requirements('requirements/runtime.txt'), |
| 128 | 'api': |
| 129 | parse_requirements('requirements/api.txt') + |
| 130 | parse_requirements('requirements/runtime.txt'), |
| 131 | 'full': |
| 132 | parse_requirements('requirements/extra.txt') + |
| 133 | parse_requirements('requirements/runtime.txt'), |
| 134 | }, |
| 135 | license='Apache License 2.0', |
| 136 | include_package_data=True, |
| 137 | packages=find_packages(), |
| 138 | keywords=[ |
| 139 | 'AI', |
| 140 | 'NLP', |
| 141 | 'in-context learning', |
| 142 | 'large language model', |
| 143 | 'evaluation', |
| 144 | 'benchmark', |
| 145 | 'llm', |
| 146 | ], |
| 147 | classifiers=[ |
| 148 | 'Programming Language :: Python :: 3.8', |
| 149 | 'Programming Language :: Python :: 3.9', |
| 150 | 'Programming Language :: Python :: 3.10', |
| 151 | 'Intended Audience :: Developers', |
| 152 | 'Intended Audience :: Education', |
| 153 | 'Intended Audience :: Science/Research', |
| 154 | ], |
| 155 | entry_points={ |
| 156 | 'console_scripts': [ |
| 157 | 'opencompass = opencompass.cli.main:main', |
| 158 | ], |
| 159 | }, |
| 160 | ) |
| 161 | |
| 162 | |
| 163 | if __name__ == '__main__': |
no test coverage detected