import numpy as np
from setuptools import setup
from Cython.Build import cythonize

setup(
    ext_modules = cythonize([
        # Model Builder
        'model_builder.pyx',

        # Data Processor
        'data_processor.pyx',

        # Post Processor
        'post_processor.pyx',

        # Trainer
        'trainer.pyx',

        # Main
        'main.pyx'
    ]),

    include_dirs = [ np.get_include() ]
)