torchio.external.imports

Complexity

Total Complexity

5

Size/Duplication

Total Lines	27
Duplicated Lines	0 %

Importance

Changes

0

4 Functions

Rating	Name	Duplication	Size	Complexity
A	get_pandas()	0	2	1
A	_check_package()	0	8	2
A	_check_and_import()	0	3	1
A	get_distinctipy()	0	2	1

from importlib import import_module
from importlib.util import find_spec
from types import ModuleType


def _check_package(*, package: str, extra: str) -> None:
    if find_spec(package) is None:
        message = (
            f'The `{package}` package is required for this.'
            f' Install TorchIO with the `{extra}` extra:'
            f' `pip install torchio[{extra}]`.'
        )
        raise ImportError(message)


def _check_and_import(package: str, extra: str) -> ModuleType:
    _check_package(package=package, extra=extra)
    return import_module(package)


def get_pandas() -> ModuleType:
    return _check_and_import(package='pandas', extra='csv')


def get_distinctipy() -> ModuleType:
    return _check_and_import(package='distinctipy', extra='plot')