From 9293ef94c8369e6adc3759ceec3d7cd513e469f4 Mon Sep 17 00:00:00 2001 From: Thomas Mansencal Date: Wed, 18 Oct 2023 18:55:26 +1300 Subject: [PATCH] Fix various typing annotations. --- colour_hdri/plotting/hdri.py | 4 +++- colour_hdri/plotting/tonemapping.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/colour_hdri/plotting/hdri.py b/colour_hdri/plotting/hdri.py index 5c769f91..b1647533 100644 --- a/colour_hdri/plotting/hdri.py +++ b/colour_hdri/plotting/hdri.py @@ -12,6 +12,8 @@ import matplotlib as mpl import matplotlib.pyplot as plt import numpy as np +from matplotlib.axes import Axes +from matplotlib.figure import Figure from colour.hints import Any, ArrayLike, Callable, Tuple from colour.plotting import CONSTANTS_COLOUR_STYLE, override_style, render @@ -38,7 +40,7 @@ def plot_HDRI_strip( ev_steps: float = -2, cctf_encoding: Callable = CONSTANTS_COLOUR_STYLE.colour.colourspace.cctf_encoding, **kwargs: Any, -) -> Tuple[plt.Figure, plt.Axes]: +) -> Tuple[Figure, Axes]: """ Plot given HDRI as strip of images of varying exposure. diff --git a/colour_hdri/plotting/tonemapping.py b/colour_hdri/plotting/tonemapping.py index 58958330..482efb1d 100644 --- a/colour_hdri/plotting/tonemapping.py +++ b/colour_hdri/plotting/tonemapping.py @@ -12,6 +12,8 @@ import matplotlib.pyplot as plt import matplotlib.ticker import numpy as np +from matplotlib.axes import Axes +from matplotlib.figure import Figure from colour.hints import Any, ArrayLike, Callable, Dict, Tuple from colour.plotting import ( @@ -41,7 +43,7 @@ def plot_tonemapping_operator_image( log_scale: bool = False, cctf_encoding: Callable = CONSTANTS_COLOUR_STYLE.colour.colourspace.cctf_encoding, **kwargs: Any, -) -> Tuple[plt.Figure, plt.Axes]: +) -> Tuple[Figure, Axes]: """ Plot given tonemapped image with superimposed luminance mapping function.