forked from bwfbowen/muax
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
24 lines (21 loc) · 812 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
import setuptools
with open('README.md', 'r') as f:
long_description = f.read()
setuptools.setup(
name='muax',
version='0.0.2.8.2',
authors = [{ 'name': "Bowen Fang", 'email': "bf2504@columbia.edu" },
{'name': 'Ian Chie', 'email': 'cc4893@columbia.edu'}],
description="A library written in Jax that provides help for using DeepMind's mctx on gym-style environments.",
long_description=long_description,
long_description_content_type='text/markdown',
packages=setuptools.find_packages(),
classifiers=[
'Programming Language :: Python :: 3',
'License :: OSI Approved :: MIT License',
'Operating System :: OS Independent'
],
python_requires='>=3.6',
py_modules=['muax'],
install_requires=['mctx', 'dm-haiku', 'optax', 'gymnasium', 'lz4', 'tensorboardX']
)