-
Notifications
You must be signed in to change notification settings - Fork 59
/
Copy pathsetup.py
32 lines (31 loc) · 1.07 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from setuptools import setup, find_packages
setup(
name='textrl',
version='0.2.22',
description='TextRL - use reinforcement learning to adjust text generation results.',
url='https://github.com/voidful/TextRL',
author='Voidful',
author_email='[email protected]',
long_description=open("README.md", encoding="utf8").read(),
long_description_content_type="text/markdown",
setup_requires=['setuptools-git'],
classifiers=[
'Development Status :: 4 - Beta',
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"License :: OSI Approved :: Apache Software License",
'Programming Language :: Python :: 3.6'
],
license="Apache",
keywords='transformer huggingface nlp generation reinforcement learning deep learning',
packages=find_packages(),
install_requires=[
"gym",
"transformers"
],
entry_points={
'console_scripts': ['textrl-dump=textrl.dump:main']
},
python_requires=">=3.5.0",
zip_safe=False,
)