setup.py

# Always prefer setuptools over distutils
from setuptools import setup, find_packages

# To use a consistent encoding
from codecs import open
from os import path

here = path.abspath(path.dirname(__file__))

# Get the long description from the README file
with open(path.join(here, 'README.md'), encoding='utf-8') as f:
    long_description = f.read()

import geomloss

setup(
    name='geomloss',
    version=geomloss.__version__,

    description='Geometric loss functions between point clouds, images and volumes.',
    long_description=long_description,
    long_description_content_type='text/markdown',

    author='Jean Feydy',
    author_email='jean.feydy@ens.fr',

    python_requires='>=3',

    classifiers=[
        'Development Status :: 2 - Pre-Alpha',

        'Intended Audience :: Developers',
        'Topic :: Scientific/Engineering',

        'License :: OSI Approved :: MIT License',

        'Operating System :: POSIX :: Linux',
        'Operating System :: MacOS :: MacOS X',
        'Operating System :: Microsoft :: Windows',

        'Programming Language :: Python :: 3 :: Only',
    ],
    keywords='kernels optimal transport measure loss geometry',

    packages=['geomloss'],

    package_data={
        'geomloss': [
        ]
    },
    scripts=[],
    url='',
    license='LICENSE.txt',
    install_requires=[
        'numpy',
    ],

    extras_require={
        'full': ['pykeops[full]'],
    },
)