Spaces:
Sleeping
Sleeping
File size: 1,356 Bytes
96da58e |
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 |
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",
"diffusers>=0.26.2",
],
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="amandlek@cs.stanford.edu",
version="0.3.0",
long_description=long_description,
long_description_content_type='text/markdown'
)
|