tools/metrics/clip-score/setup.py

import os

import setuptools


def read(rel_path):
    base_path = os.path.abspath(os.path.dirname(__file__))
    with open(os.path.join(base_path, rel_path)) as f:
        return f.read()


def get_version(rel_path):
    for line in read(rel_path).splitlines():
        if line.startswith("__version__"):
            delim = '"' if '"' in line else "'"
            return line.split(delim)[1]

    raise RuntimeError("Unable to find version string.")


if __name__ == "__main__":
    setuptools.setup(
        name="clip-score",
        version=get_version(os.path.join("src", "clip_score", "__init__.py")),
        author="Taited",
        author_email="[email protected]",
        description=("Package for calculating CLIP-Score" " using PyTorch"),
        long_description=read("README.md"),
        long_description_content_type="text/markdown",
        url="https://github.com/taited/clip-score",
        package_dir={"": "src"},
        packages=setuptools.find_packages(where="src"),
        classifiers=[
            "Programming Language :: Python :: 3",
            "License :: OSI Approved :: Apache Software License",
        ],
        python_requires=">=3.5",
        entry_points={
            "console_scripts": [
                "clip-score = clip_score.clip_score:main",
            ],
        },
        install_requires=[
            "numpy",
            "pillow",
            "torch>=1.7.1",
            "torchvision>=0.8.2",
            "ftfy",
            "regex",
            "tqdm",
        ],
        extras_require={"dev": ["flake8", "flake8-bugbear", "flake8-isort", "nox"]},
    )