-
Notifications
You must be signed in to change notification settings - Fork 195
/
setup.py
25 lines (23 loc) · 917 Bytes
/
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
import setuptools
import pathlib
setuptools.setup(
name='dreamerv2',
version='2.2.0',
description='Mastering Atari with Discrete World Models',
url='http://github.com/danijar/dreamerv2',
long_description=pathlib.Path('README.md').read_text(),
long_description_content_type='text/markdown',
packages=['dreamerv2', 'dreamerv2.common'],
package_data={'dreamerv2': ['configs.yaml']},
entry_points={'console_scripts': ['dreamerv2=dreamerv2.train:main']},
install_requires=[
'gym[atari]', 'atari_py', 'crafter', 'dm_control', 'ruamel.yaml',
'tensorflow', 'tensorflow_probability'],
classifiers=[
'Intended Audience :: Science/Research',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python :: 3',
'Topic :: Games/Entertainment',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
],
)