setup.py

from setuptools import setup, find_packages
import sys, os.path

# Don't import module here, since deps may not be installed
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'airl'))
from version import VERSION

setup(
    name='adversarial-irl',
    version=VERSION,
    packages=find_packages(exclude=['scripts', 'tabular_maxent_irl']),
    package_data={
        'envs': ['*.xml'],
    },
    install_requires=[
        # 'rllab@git+https://github.com/shwang/rllab.git',
        # just call pip install git+https://github.com/shwang/rllab.git
        'mujoco_py',
    ],

    # Metadata
    author='Justin Fu, Adam Gleave',
    license='GPL 3.0',
    description='Implementation of adversarial IRL (Fu et al, 2017)',
    url='https://github.com/AdamGleave/inverse_rl',
)