File size: 414 Bytes
e8ac98a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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() ]
)