forked from huawei-noah/HEBO
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
35 lines (33 loc) · 1.13 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
# Created by Paul Daoudi
# Date: 11/02/2023
from setuptools import setup, find_packages
setup(author='Paul Daoudi',
name='rllg',
version='0.1.0',
packages=find_packages(include=['my_package', 'my_package.*']),
install_requires=[
'setuptools==65.5.0',
'wheel==0.38.0',
'numpy==1.23.1',
'torch==1.10.2',
'tensorboardX==2.4.1',
'mujoco-py==2.1.2.14',
'omegaconf==2.1.1',
'protobuf==3.20.0',
# Install gym by hand. Works with traditional pip install -e . inside a conda env,
# but problem with docker otherwise.
# 'gym==0.21.0',
'ray[tune]==1.9.2',
'pyyaml',
'matplotlib',
'ipython',
'pandas',
'matplotlib',
'jupyter',
'ml-collections',
'scipy',
# Install dmc2gym by hand, with git clone git+https://github.com/denisyarats/dmc2gym.git and
# pip install -e .
'dmc2gym @ git+https://github.com/denisyarats/dmc2gym.git'
]
)