setup.py

import os.path

from setuptools import setup

# Store simulators path before installing
with open(os.path.join(os.path.dirname(__file__), "uitb/utils/__simulatorsdir__.py"), "w") as f:
    f.write("# This file is read-only and should not be modified manually!\n"
            "# To use a different output directory, set the 'simulator_folder' tag in the config file.\n\n")
    f.write("SIMULATORS_DIR = " + repr(os.path.normpath(os.path.abspath('simulators'))))

setup(
   name='uitb',
   version='2.0.1',
   author='Aleksi Ikkala',
   author_email='aleksi.ikkala@gmail.com',
   packages=['uitb'],
   package_data={'': ['**']},
   url='https://github.com/aikkala/user-in-the-box',
   license='LICENSE',
   description='Modeling and simulating HCI interaction tasks in MuJoCo',
   long_description=open('README.md').read(),
   python_requires='>=3.8',
   install_requires=[
       "gymnasium>=0.28",
       "pygame",
       "pynput",
       "mujoco==2.3.5",
       "imageio",
       "stable_baselines3==2.2.1",
       "torch",
       "wandb", "tensorboard",
       "numpy", "matplotlib", "scipy",
       "opencv-python",
       "ruamel.yaml",
       "zmq"
   ],
)