setup.py

from setuptools import setup

with open("README.md") as fh:
    long_description = ""
    header_count = 0
    for line in fh:
        if line.startswith("##"):
            header_count += 1
        if header_count < 2:
            long_description += line
        else:
            break

# pytest is pinned to 7.0.1 as this is last version for python 3.6
extras = {"testing": ["pytest==7.0.1"]}

setup(
    name="mini_behavior",
    author="Stanford & UT Austin",
    author_email="jiahengh@utexas.edu",
    classifiers=[
        "Development Status :: 5 - Production/Stable",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
    ],
    version="1.1.0",
    keywords="memory, environment, agent, rl, gym",
    url="https://github.com/StanfordVL/mini_behavior",
    description="Minimalistic gridworld reinforcement learning environments",
    extras_require=extras,
    packages=["mini_behavior", "mini_behavior.envs"],
    license="Apache",
    long_description=long_description,
    long_description_content_type="text/markdown",
    install_requires=[
        "gym==0.21",
        "numpy>=1.18.0",
        "matplotlib>=3.0",
        "h5py",
    ],
    python_requires=">=3.7",
    tests_require=extras["testing"],
)