#!/usr/bin/env python
from distutils.core import setup
from catkin_pkg.python_setup import generate_distutils_setup

setup_args = generate_distutils_setup(
    packages=['ros_message_test'],
    scripts=['scripts/rosmessage_test.py'],
    package_dir={'': 'scripts'},
)

setup(**setup_args, requires=['matplotlib', 'numpy', 'PIL', 'torch', 'torchvision', 'gym'])
