56 строки
1.6 KiB
Python
56 строки
1.6 KiB
Python
# Copyright (c) Microsoft Corporation
|
|
# Licensed under the MIT License.
|
|
|
|
import setuptools
|
|
|
|
# Version will be read from version.py
|
|
version = ''
|
|
name = 'responsibleai-text'
|
|
# Fetch Version
|
|
with open('responsibleai_text/version.py') as f:
|
|
code = compile(f.read(), f.name, 'exec')
|
|
exec(code)
|
|
|
|
# Fetch ReadMe
|
|
with open('README.md', 'r') as fh:
|
|
long_description = fh.read()
|
|
|
|
# Use requirements.txt to set the install_requires
|
|
with open('requirements.txt') as f:
|
|
install_requires = [line.strip() for line in f]
|
|
EXTRAS = {
|
|
"qa": [
|
|
'evaluate',
|
|
'bert_score',
|
|
'nltk',
|
|
'rouge_score'
|
|
],
|
|
"generative_text": [
|
|
'interpret_text',
|
|
'sentence_transformers'
|
|
]
|
|
}
|
|
setuptools.setup(
|
|
name=name, # noqa: F821
|
|
version=version, # noqa: F821
|
|
author="Roman Lutz, Ilya Matiach, Ke Xu",
|
|
author_email="raiwidgets-maintain@microsoft.com",
|
|
description="SDK API to assess text "
|
|
"Machine Learning models.",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/microsoft/responsible-ai-toolbox",
|
|
packages=setuptools.find_packages(),
|
|
python_requires='>=3.7',
|
|
extras_require=EXTRAS,
|
|
install_requires=install_requires,
|
|
classifiers=[
|
|
"Programming Language :: Python :: 3.7",
|
|
"Programming Language :: Python :: 3.8",
|
|
"Programming Language :: Python :: 3.9",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Operating System :: OS Independent",
|
|
"Development Status :: 3 - Alpha"
|
|
]
|
|
)
|