setup.py

import io
import os
import re
from os import path

from setuptools import find_packages
from setuptools import setup


this_directory = path.abspath(path.dirname(__file__))
with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
    long_description = f.read()


setup(
    name="mario-gpt",
    version="0.1.0",
    url="https://github.com/shyamsn97/mario-gpt",
    license='MIT',

    author="Shyam Sudhakaran",
    author_email="shyamsnair@protonmail.com",

    description="Generating Mario Levels with GPT2. Code for the paper: 'MarioGPT: Open-Ended Text2Level Generation through Large Language Models', https://arxiv.org/abs/2302.05981",

    long_description=long_description,
    long_description_content_type="text/markdown",

    packages=find_packages(exclude=('tests',)),

    install_requires=[
        'torch',
        'transformers',
        'scipy',
        'tqdm',
        'pillow',
        'numpy',
    ],

    extras_require={
        'dev': [
            'pytest',
            'black',
            'flake8',
            'isort',
        ],
        'gradio': [
            'gradio>=5.48.0',
            'fastapi',
            'uvicorn[standard]',
            'spaces',
        ],
        'mcp': [
            'mcp>=0.9.0',
            'pydantic>=2.0.0',
        ],
    },

    classifiers=[
        'Development Status :: 3 - Alpha',
        'License :: OSI Approved :: MIT License',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.8',
        'Programming Language :: Python :: 3.9',
        'Programming Language :: Python :: 3.10',
        'Topic :: Scientific/Engineering :: Artificial Intelligence',
        'Topic :: Games/Entertainment',
    ],
    
    python_requires='>=3.8',
)