setup.py

import os
import sys
import shutil
from setuptools import setup


def clean_repo():
    repo_folder = os.path.realpath(os.path.dirname(__file__))
    dist_folder = os.path.join(repo_folder, "dist")
    build_folder = os.path.join(repo_folder, "build")
    if os.path.isdir(dist_folder):
        shutil.rmtree(dist_folder, ignore_errors=True)
    if os.path.isdir(build_folder):
        shutil.rmtree(build_folder, ignore_errors=True)


# Read version string
_version = None
script_folder = os.path.realpath(os.path.dirname(__file__))
with open(os.path.join(script_folder, "ibug", "age_estimation", "__init__.py")) as init:
    for line in init.read().splitlines():
        fields = (
            line.replace("=", " ")
            .replace("'", " ")
            .replace('"', " ")
            .replace("\t", " ")
            .split()
        )
        if len(fields) >= 2 and fields[0] == "__version__":
            _version = fields[1]
            break
if _version is None:
    sys.exit("Sorry, cannot find version information.")

# Installation
config = {
    "name": "age_estimation",
    "version": _version,
    "description": "FPAge",
    "author": "Yiming Lin",
    "author_email": "yiming.lin15@imperial.ac.uk",
    "packages": ["ibug.age_estimation"],
    "install_requires": [
        "numpy>=1.17.0",
        "scipy>=1.1.0",
        "torch>=1.6.0",
        "torchvision>=0.7.0",
        "opencv-python>= 3.4.2",
        "timm>=0.4.12",
    ],
    "zip_safe": False,
}
clean_repo()
setup(**config)
clean_repo()