Function prompt_install

modelcache_mm/utils/dependency_control.py:7–18

Function used to prompt user to install a package.

(package: str, warn: bool = False)

def prompt_install(package: str, warn: bool = False):  # pragma: no cover
    """
    Function used to prompt user to install a package.
    """
    cmd = f"pip install {package}"
    try:
        if warn and input(f"Install {package}? Y/n: ") != "Y":
            raise ModuleNotFoundError(f"No module named {package}")
        subprocess.check_call(cmd, shell=True)
        modelcache_log.info("%s installed successfully!", package)
    except subprocess.CalledProcessError as e:
        raise PipInstallError(package) from e