from distutils.core import setup
from distutils.extension import Extension
from Cython.Build import cythonize
import numpy
extensions = [
Extension(
"im2col_cython", ["im2col_cython.pyx"], include_dirs=[numpy.get_include()]
),
]
setup(ext_modules=cythonize(extensions),)