forked from mihdalal/robomimic
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
43 lines (40 loc) · 1.29 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
from setuptools import setup, find_packages
# read the contents of your README file
from os import path
this_directory = path.abspath(path.dirname(__file__))
with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
lines = f.readlines()
# remove images from README
lines = [x for x in lines if (('.png' not in x) and ('.gif' not in x))]
long_description = ''.join(lines)
setup(
name="robomimic",
packages=[
package for package in find_packages() if package.startswith("robomimic")
],
install_requires=[
"numpy>=1.13.3",
"h5py",
"psutil",
"tqdm",
"termcolor",
"tensorboard",
"tensorboardX",
"imageio",
"imageio-ffmpeg",
"matplotlib",
"egl_probe>=1.0.1",
"torch",
"torchvision",
],
eager_resources=['*'],
include_package_data=True,
python_requires='>=3',
description="robomimic: A Modular Framework for Robot Learning from Demonstration",
author="Ajay Mandlekar, Danfei Xu, Josiah Wong, Soroush Nasiriany, Chen Wang, Matthew Bronars",
url="https://github.com/ARISE-Initiative/robomimic",
author_email="[email protected]",
version="0.3.0",
long_description=long_description,
long_description_content_type='text/markdown'
)