Fix various typing annotations.

colour-science · Oct 18, 2023 · 9293ef9 · 9293ef9
1 parent 2ec9dc3
commit 9293ef9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -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.