Skip to content
Snippets Groups Projects
Commit 087fbfa3 authored by zqwerty's avatar zqwerty
Browse files

Merge branch 'update_env' into unified_dataset

parents 7ae31c8b 315dae17
No related branches found
No related tags found
No related merge requests found
''' '''
setup.py for ConvLab-2 setup.py for ConvLab-3
''' '''
import sys import sys
import os import os
...@@ -11,57 +11,48 @@ class LibTest(TestCommand): ...@@ -11,57 +11,48 @@ class LibTest(TestCommand):
def run_tests(self): def run_tests(self):
# import here, cause outside the eggs aren't loaded # import here, cause outside the eggs aren't loaded
ret = os.system("pytest --cov=ConvLab-2 tests/ --cov-report term-missing") ret = os.system("pytest --cov=ConvLab-3 tests/ --cov-report term-missing")
sys.exit(ret >> 8) sys.exit(ret >> 8)
setup( setup(
name='ConvLab-2', name='ConvLab',
version='1.0.0', version='3.0.0',
packages=find_packages(exclude=[]), packages=find_packages(exclude=[]),
license='Apache', license='Apache',
description='Task-oriented Dialog System Toolkits', description='An Open-source Dialog System Toolkits',
long_description=open('README.md', encoding='UTF-8').read(), long_description=open('README.md', encoding='UTF-8').read(),
long_description_content_type="text/markdown", long_description_content_type="text/markdown",
classifiers=[ classifiers=[
'Development Status :: 2 - Pre-Alpha', 'Development Status :: 3 - Alpha',
'License :: OSI Approved :: Apache Software License', 'License :: OSI Approved :: Apache Software License',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6', 'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Intended Audience :: Science/Research', 'Intended Audience :: Science/Research',
'Topic :: Scientific/Engineering :: Artificial Intelligence', 'Topic :: Scientific/Engineering :: Artificial Intelligence',
], ],
install_requires=[ install_requires=[
'nltk>=3.4', 'tabulate',
'tqdm>=4.30', 'python-Levenshtein',
'checksumdir>=1.1',
'dataclasses',
'visdom',
'Pillow',
'future',
'torch>=1.2.0,<1.6.0',
'numpy>=1.15.0',
'scipy',
'scikit-learn==0.20.3',
'pytorch-pretrained-bert>=0.6.1',
'transformers>=2.3.0,<3.0.0',
'tensorboard>=1.14.0',
'tensorboardX==1.7',
'tokenizers>=0.8.0',
'overrides==4.1.2',
'allennlp==0.9.0',
'requests', 'requests',
'numpy',
'nltk',
'scipy',
'torch>=1.6',
'transformers>=4.0',
'spacy',
'allennlp',
'simplejson', 'simplejson',
'spacy==2.1.9',
'unidecode', 'unidecode',
'jieba', 'jieba',
'embeddings', 'embeddings',
'visdom',
'quadprog', 'quadprog',
'pyyaml',
'fuzzywuzzy', 'fuzzywuzzy',
'python-Levenshtein',
'json_lines', 'json_lines',
'gtts', 'gtts',
'DeepSpeech', 'deepspeech',
'pydub' 'pydub'
], ],
extras_require={ extras_require={
...@@ -78,15 +69,11 @@ setup( ...@@ -78,15 +69,11 @@ setup(
] ]
}, },
cmdclass={'test': LibTest}, cmdclass={'test': LibTest},
entry_points={ entry_points={},
'console_scripts': [
"ConvLab-2-report=convlab2.scripts:report"
]
},
include_package_data=True, include_package_data=True,
url='https://github.com/thu-coai/ConvLab-2', url='https://github.com/ConvLab/ConvLab-3',
author='thu-coai', author='thu-coai',
author_email='thu-coai-developer@googlegroups.com', author_email='thu-coai-developer@googlegroups.com',
python_requires='>=3.5', python_requires='>=3.6',
zip_safe=False zip_safe=False
) )
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment