sklearn/utils/_optional_dependencies.py - main - githubmirror/scikit-learn

sklearn/utils/_optional_dependencies.py

62 строки2 KB

Jérémie du Boisberranger

FEA Callbacks API (#33322)

19 май 2026, 11:34

19 май 2026, 11:34f22c08e

# Authors: The scikit-learn developers
# SPDX-License-Identifier: BSD-3-Clause


def check_matplotlib_support(caller_name):
    """Raise ImportError with detailed error message if mpl is not installed.

    Plot utilities like any of the Display's plotting functions should lazily import
    matplotlib and call this helper before any computation.

    Parameters
    ----------
    caller_name : str
        The name of the caller that requires matplotlib.
    """
    try:
        import matplotlib  # noqa: F401
    except ImportError as e:
        raise ImportError(
            f"{caller_name} requires matplotlib. You can install matplotlib with "
            "`pip install matplotlib`"
        ) from e


def check_pandas_support(caller_name):
    """Raise ImportError with detailed error message if pandas is not installed.

    Plot utilities like :func:`fetch_openml` should lazily import
    pandas and call this helper before any computation.

    Parameters
    ----------
    caller_name : str
        The name of the caller that requires pandas.

    Returns
    -------
    pandas
        The pandas package.
    """
    try:
        import pandas

        return pandas
    except ImportError as e:
        raise ImportError(f"{caller_name} requires pandas.") from e


def check_rich_support(caller_name):
    """Raise ImportError with detailed error message if rich is not installed.

    Caller should lazily import rich and call this helper before any computation.

    Parameters
    ----------
    caller_name : str
        The name of the caller that requires rich.
    """
    try:
        import rich  # noqa: F401
    except ImportError as e:
        raise ImportError(f"{caller_name} requires rich.") from e