2019-01-31 23:34:32 -05:00
|
|
|
"""
|
|
|
|
rltorch stands for Reinforcement Learning Torch -- RL library built on top of PyTorch
|
|
|
|
"""
|
|
|
|
import setuptools
|
|
|
|
|
|
|
|
|
|
|
|
setuptools.setup(
|
|
|
|
name="rltorch",
|
|
|
|
author="Brandon Rozek",
|
|
|
|
author_email="rozekbrandon@gmail.com",
|
|
|
|
license='MIT',
|
|
|
|
description="Reinforcement Learning Framework for PyTorch",
|
|
|
|
version="0.1",
|
|
|
|
packages=setuptools.find_packages(),
|
2020-03-15 14:27:56 -04:00
|
|
|
install_requires=[
|
|
|
|
"numpy~=1.16.0",
|
|
|
|
"opencv-python~=4.2.0.32",
|
|
|
|
"gym~=0.10.11",
|
|
|
|
"torch~=1.4.0",
|
|
|
|
"numba~=0.48.0"
|
|
|
|
]
|
|
|
|
)
|