setup.py

from setuptools import setup, find_packages
from setuptools.command.install import install


__version__ = "1.2.2"

with open('README.md', 'r') as readme_file:
    long_description = readme_file.read()


class CustomInstall(install):
    def run(self):
        install.run(self)

setup(
    name='rlgym-ppo',
    packages=find_packages(),
    version=__version__,
    description='A multi-processed implementation of PPO for use with RLGym.',
    long_description=long_description,
    long_description_content_type='text/markdown',
    author='Matthew Allen',
    url='https://github.com/AechPro/rlgym-ppo',

    install_requires=[
        'cloudpickle>2.0',
        'filelock>3.0',
        'gym>0.23',
        'gym-notices==0.0.8',
        'Jinja2>3.0',
        'mpmath>1.0',
        'numpy>1.21',
        'sympy>1.10',
        'torch>1.13',
        'wandb>0.15',
        'typing_extensions>4.6'
    ],
    python_requires='>=3.7',
    cmdclass={'install': CustomInstall},
    license='Apache 2.0',
    license_file='LICENSE',
    keywords=['rocket-league', 'gym', 'reinforcement-learning', 'simulation', 'ppo', 'rlgym', 'rocketsim']
)