setup.py

from setuptools import setup, find_packages

setup(name='dial-mpc',
      author="Haoru Xue",
      author_email="haoru-xue@berkeley.edu",
      packages=find_packages(include="dial_mpc"),
      version='0.0.2',
      install_requires=[
          'numpy<2.0.0',
          'matplotlib',
          'tqdm',
          'tyro',
          'jax[cuda12]',
          'jax-cosmo',
          'mujoco<=3.2.4',
          'brax',
          'art',
          'emoji',
          'scienceplots'
      ],
      package_data={'dial-mpc': ['models/', 'examples/']},
      entry_points={
          'console_scripts': [
              'dial-mpc=dial_mpc.core.dial_core:main',
              'dial-mpc-sim2sim=dial_mpc.core.dial_sim2sim:main',
              'dial-mpc-sim2real=dial_mpc.core.dial_sim2real:main',
              'dial-mpc-sim=dial_mpc.deploy.dial_sim:main',
              'dial-mpc-real=dial_mpc.deploy.dial_real:main',
              'dial-mpc-plan=dial_mpc.deploy.dial_plan:main',
          ],
      },
      )