diff --git a/setup.py b/setup.py index 321a111f8971eef619c14642320e145d108dbcea..bb24ca3037a6016759e470fff5764eff9e3e219d 100755 --- a/setup.py +++ b/setup.py @@ -1,5 +1,5 @@ ''' -setup.py for ConvLab-2 +setup.py for ConvLab-3 ''' import sys import os @@ -11,57 +11,48 @@ class LibTest(TestCommand): def run_tests(self): # 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) setup( - name='ConvLab-2', - version='1.0.0', + name='ConvLab', + version='3.0.0', packages=find_packages(exclude=[]), 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_content_type="text/markdown", classifiers=[ - 'Development Status :: 2 - Pre-Alpha', + 'Development Status :: 3 - Alpha', 'License :: OSI Approved :: Apache Software License', - 'Programming Language :: Python :: 3.5', 'Programming Language :: Python :: 3.6', + 'Programming Language :: Python :: 3.7', + 'Programming Language :: Python :: 3.8', + 'Programming Language :: Python :: 3.9', 'Intended Audience :: Science/Research', 'Topic :: Scientific/Engineering :: Artificial Intelligence', ], install_requires=[ - 'nltk>=3.4', - 'tqdm>=4.30', - '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', + 'tabulate', + 'python-Levenshtein', 'requests', + 'numpy', + 'nltk', + 'scipy', + 'torch>=1.6', + 'transformers>=4.0', + 'spacy', + 'allennlp', 'simplejson', - 'spacy==2.1.9', 'unidecode', 'jieba', 'embeddings', + 'visdom', 'quadprog', - 'pyyaml', 'fuzzywuzzy', - 'python-Levenshtein', 'json_lines', 'gtts', - 'DeepSpeech', + 'deepspeech', 'pydub' ], extras_require={ @@ -78,15 +69,11 @@ setup( ] }, cmdclass={'test': LibTest}, - entry_points={ - 'console_scripts': [ - "ConvLab-2-report=convlab2.scripts:report" - ] - }, + entry_points={}, include_package_data=True, - url='https://github.com/thu-coai/ConvLab-2', + url='https://github.com/ConvLab/ConvLab-3', author='thu-coai', author_email='thu-coai-developer@googlegroups.com', - python_requires='>=3.5', + python_requires='>=3.6', zip_safe=False )