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]"],
    },
)