forked from zjowowen/GenerativeRL_Preview
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
34 lines (32 loc) · 746 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
26
27
28
29
30
31
32
33
34
from setuptools import setup, find_packages
setup(
name='GenerativeRL',
version='0.0.1',
description='PyTorch implementations of generative reinforcement learning algorithms',
author='zjowowen',
packages=find_packages(
exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
install_requires=[
'gym',
'numpy',
'torch>=2.2.0',
'opencv-python',
'tensordict',
'matplotlib',
'wandb',
'rich',
'mujoco_py',
'easydict',
'Cython<3.0',
'tqdm',
'torchdyn',
'torchode',
'torchsde',
'scipy',
'diffusers',
'timm',
'av',
'moviepy',
'imageio[ffmpeg]',
]
)