Add typing block for inspection

Author: cvanelteren

ultraplot/__init__.py

@@ -7,10 +7,135 @@
 
 import sys
 from pathlib import Path
-from typing import Optional
+from typing import TYPE_CHECKING, Optional
 
 from ._lazy import LazyLoader, install_module_proxy
 
+if TYPE_CHECKING:
+    # These imports are never executed at runtime, so they have zero effect on
+    # import performance. They exist solely so that type checkers (pyright, mypy)
+    # can resolve names that are otherwise provided by the lazy loader at runtime.
+    #
+    # Keep this block in sync with _LAZY_LOADING_EXCEPTIONS and every submodule's
+    # __all__ — that is the only maintenance burden.
+    import matplotlib.pyplot as pyplot
+
+    from .axes import Axes as Axes
+    from .axes import CartesianAxes as CartesianAxes
+    from .axes import ExternalAxesContainer as ExternalAxesContainer
+    from .axes import GeoAxes as GeoAxes
+    from .axes import PlotAxes as PlotAxes
+    from .axes import PolarAxes as PolarAxes
+    from .axes import ThreeAxes as ThreeAxes
+    from .colors import ColormapDatabase as ColormapDatabase
+    from .colors import ColorDatabase as ColorDatabase
+    from .colors import ContinuousColormap as ContinuousColormap
+    from .colors import DiscreteColormap as DiscreteColormap
+    from .colors import DiscreteNorm as DiscreteNorm
+    from .colors import DivergingNorm as DivergingNorm
+    from .colors import LinearSegmentedColormap as LinearSegmentedColormap
+    from .colors import LinearSegmentedNorm as LinearSegmentedNorm
+    from .colors import ListedColormap as ListedColormap
+    from .colors import PerceptualColormap as PerceptualColormap
+    from .colors import PerceptuallyUniformColormap as PerceptuallyUniformColormap
+    from .colors import SegmentedNorm as SegmentedNorm
+    from .colors import _cmap_database as colormaps
+    from .config import Configurator as Configurator
+    from .config import rc as rc
+    from .config import rc_matplotlib as rc_matplotlib
+    from .config import rc_ultraplot as rc_ultraplot
+    from .config import use_style as use_style
+    from .constructor import Colormap as Colormap
+    from .constructor import Colors as Colors
+    from .constructor import Cycle as Cycle
+    from .constructor import Formatter as Formatter
+    from .constructor import FORMATTERS as FORMATTERS
+    from .constructor import Locator as Locator
+    from .constructor import LOCATORS as LOCATORS
+    from .constructor import Norm as Norm
+    from .constructor import NORMS as NORMS
+    from .constructor import Proj as Proj
+    from .constructor import PROJS as PROJS
+    from .constructor import Scale as Scale
+    from .constructor import SCALES as SCALES
+    from .demos import show_channels as show_channels
+    from .demos import show_cmaps as show_cmaps
+    from .demos import show_colorspaces as show_colorspaces
+    from .demos import show_colors as show_colors
+    from .demos import show_cycles as show_cycles
+    from .demos import show_fonts as show_fonts
+    from .figure import Figure as Figure
+    from .gridspec import GridSpec as GridSpec
+    from .gridspec import SubplotGrid as SubplotGrid
+    from .proj import Aitoff as Aitoff
+    from .proj import Hammer as Hammer
+    from .proj import KavrayskiyVII as KavrayskiyVII
+    from .proj import NorthPolarAzimuthalEquidistant as NorthPolarAzimuthalEquidistant
+    from .proj import NorthPolarGnomonic as NorthPolarGnomonic
+    from .proj import NorthPolarLambertAzimuthalEqualArea as NorthPolarLambertAzimuthalEqualArea
+    from .proj import SouthPolarAzimuthalEquidistant as SouthPolarAzimuthalEquidistant
+    from .proj import SouthPolarGnomonic as SouthPolarGnomonic
+    from .proj import SouthPolarLambertAzimuthalEqualArea as SouthPolarLambertAzimuthalEqualArea
+    from .proj import WinkelTripel as WinkelTripel
+    from .scale import CutoffScale as CutoffScale
+    from .scale import ExpScale as ExpScale
+    from .scale import FuncScale as FuncScale
+    from .scale import InverseScale as InverseScale
+    from .scale import LinearScale as LinearScale
+    from .scale import LogitScale as LogitScale
+    from .scale import LogScale as LogScale
+    from .scale import MercatorLatitudeScale as MercatorLatitudeScale
+    from .scale import PowerScale as PowerScale
+    from .scale import SineLatitudeScale as SineLatitudeScale
+    from .scale import SymmetricalLogScale as SymmetricalLogScale
+    from .text import CurvedText as CurvedText
+    from .ticker import AutoCFDatetimeFormatter as AutoCFDatetimeFormatter
+    from .ticker import AutoCFDatetimeLocator as AutoCFDatetimeLocator
+    from .ticker import AutoFormatter as AutoFormatter
+    from .ticker import CFDatetimeFormatter as CFDatetimeFormatter
+    from .ticker import DegreeFormatter as DegreeFormatter
+    from .ticker import DegreeLocator as DegreeLocator
+    from .ticker import DiscreteLocator as DiscreteLocator
+    from .ticker import FracFormatter as FracFormatter
+    from .ticker import IndexFormatter as IndexFormatter
+    from .ticker import IndexLocator as IndexLocator
+    from .ticker import LatitudeFormatter as LatitudeFormatter
+    from .ticker import LatitudeLocator as LatitudeLocator
+    from .ticker import LongitudeFormatter as LongitudeFormatter
+    from .ticker import LongitudeLocator as LongitudeLocator
+    from .ticker import SciFormatter as SciFormatter
+    from .ticker import SigFigFormatter as SigFigFormatter
+    from .ticker import SimpleFormatter as SimpleFormatter
+    from .ui import close as close
+    from .ui import figure as figure
+    from .ui import ioff as ioff
+    from .ui import ion as ion
+    from .ui import isinteractive as isinteractive
+    from .ui import show as show
+    from .ui import subplot as subplot
+    from .ui import subplots as subplots
+    from .ui import switch_backend as switch_backend
+    from .utils import arange as arange
+    from .utils import check_for_update as check_for_update
+    from .utils import edges as edges
+    from .utils import edges2d as edges2d
+    from .utils import get_colors as get_colors
+    from .utils import saturate as saturate
+    from .utils import scale_luminance as scale_luminance
+    from .utils import scale_saturation as scale_saturation
+    from .utils import set_alpha as set_alpha
+    from .utils import set_hue as set_hue
+    from .utils import set_luminance as set_luminance
+    from .utils import set_saturation as set_saturation
+    from .utils import shade as shade
+    from .utils import shift_hue as shift_hue
+    from .utils import to_hex as to_hex
+    from .utils import to_rgb as to_rgb
+    from .utils import to_rgba as to_rgba
+    from .utils import to_xyz as to_xyz
+    from .utils import to_xyza as to_xyza
+    from .utils import units as units
+
 name = "ultraplot"
 
 try: