_typing.pyi

# This file is generated by docs2stub. These types are intended
# to simplify the stubs generated by docs2stub. They are not
# intended to be used directly by other users.

import io
import typing_extensions
from decimal import Decimal as Decimal
from typing import Any

import numpy as np
import numpy.typing as npt
import pandas as pd
from numpy.typing import ArrayLike as ArrayLike
from scipy.sparse import spmatrix

from .base import BaseEstimator, ClassifierMixin, RegressorMixin

PythonScalar: typing_extensions.TypeAlias = str | int | float | bool

NDArray: typing_extensions.TypeAlias = npt.NDArray[Any]
MatrixLike: typing_extensions.TypeAlias = NDArray | pd.DataFrame | spmatrix
FileLike = io.IOBase
PathLike = str
Int: typing_extensions.TypeAlias = int | np.int8 | np.int16 | np.int32 | np.int64
Float: typing_extensions.TypeAlias = float | np.float16 | np.float32 | np.float64

PandasScalar: typing_extensions.TypeAlias = pd.Period | pd.Timestamp | pd.Timedelta | pd.Interval
Scalar: typing_extensions.TypeAlias = PythonScalar | PandasScalar

Estimator = BaseEstimator
Classifier = ClassifierMixin
Regressor = RegressorMixin

Color: typing_extensions.TypeAlias = tuple[float, float, float] | str