setup.py

from os.path import join, dirname, realpath
from setuptools import setup, find_packages
from setuptools.dist import Distribution
import sys

assert sys.version_info.major == 3 and sys.version_info.minor >= 6, \
    "This repo is designed to work with Python 3.6 and greater." \
    + "Please install it before proceeding."


class BinaryDistribution(Distribution):
    """Distribution which always forces a binary package with platform name"""
    def has_ext_modules(foo):
        return True


setup(
    name='teachDRL',
    
    packages=find_packages(),
    version="0.1.0.dev0",
    install_requires=[
        # 'cloudpickle==1.2.0',
        'gym[atari,box2d,classic_control]>=0.10.8',
        # 'ipython',
        # 'joblib',
        # 'matplotlib',
        'numpy',
        # 'pandas',
        # 'pytest',
        'psutil',
        # 'scipy',
        'sklearn',
        # 'imageio',
        # 'seaborn==0.8.1',
        # 'tensorflow<=1.16.0',
        # 'setuptools',
        # 'treelib',
        # 'gizeh',
        # 'tqdm'
    ],
    description="Teacher algorithms for curriculum learning of Deep RL in continuously parameterized environments",
    author="Rémy Portelas",
    # Uncomment this is you need platform specific wheels
    # distclass=BinaryDistribution,
)