-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
46 lines (43 loc) · 1.19 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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#!/usr/bin/env python
from setuptools import setup, find_packages
required = [
'flax>=0.4.1',
'jax>=0.4.13',
'jaxtyping>=0.2.20',
'jaxlib>=0.4.13',
'pytest>=7.4.0',
'matplotlib>=3.5.1',
'numpy>=1.22.2',
'optax>=0.1.1',
'scipy>=1.8.0',
'wandb>=0.12.11',
'argparse-dataclass>=0.2.1',
'jaxutils',
'chex',
'brax',
'distrax @ git+https://github.com/deepmind/distrax.git',
'trajax @ git+https://[email protected]/lenarttreven/trajax.git',
'bsm @ git+https://github.com/lasgroup/bayesian_statistical_models.git',
]
extras = {}
setup(
name='mbpo',
version='0.0.2',
license="MIT",
packages=find_packages(),
python_requires='>=3.9',
install_requires=required,
extras_require=extras,
classifiers=[
# How mature is this project? Common values are
# 3 - Alpha
# 4 - Beta
# 5 - Production/Stable
"Development Status :: 3 - Alpha",
# Indicate who your project is intended for
"Intended Audience :: Developers",
"Topic :: Software Development :: Build Tools",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3.11",
],
)