setup.py

import os
import platform

from setuptools import setup, find_packages, Extension

try:
    from Cython.Build import cythonize
except:
    print ('You must have Cython installed. '
           'Run sudo pip install cython to do so.')
    raise


# Use gcc for openMP on OSX
if 'darwin' in platform.platform().lower():
    os.environ["CC"] = "gcc-4.9"
    os.environ["CXX"] = "g++-4.9"


# Declare extension
extensions = [Extension("glove.glove_cython", ["glove/glove_cython.pyx"],
                        extra_link_args=["-fopenmp"],
                        extra_compile_args=['-fopenmp']),
              Extension("glove.metrics.accuracy_cython",
                        ["glove/metrics/accuracy_cython.pyx"],
                        extra_link_args=["-fopenmp"],
                        extra_compile_args=['-fopenmp']),
              Extension("glove.corpus_cython", ["glove/corpus_cython.pyx"],
                        language='C++',
                        extra_compile_args=['-std=c++11', '-O3'])]

setup(
    name='glove',
    version='0.0.1',
    description=('Python implementation of Global Vectors '
                 'for Word Representation (GloVe)'),
    long_description='',
    packages=["glove"],
    install_requires=['numpy',
                      'cython',
                      'scipy'],
    author='Maciej Kula',
    url='https://github.com/maciejkula/glove-python',
    license='Apache 2.0',
    classifiers=['Development Status :: 3 - Alpha'],
    ext_modules=cythonize(extensions)
)