fix error tox Union

Ouranosinc · Sep 25, 2024 · bff8648 · bff8648
1 parent c956d60
commit bff8648
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/figanos/hvplot/plot.py b/src/figanos/hvplot/plot.py
@@ -4,7 +4,7 @@
 import warnings
 from functools import partial
 from pathlib import Path
-from typing import Any
+from typing import Any, Union
 
 import holoviews as hv
 import hvplot  # noqa: F401
@@ -145,7 +145,7 @@ def _plot_ens_pct_stats(
 
 def _plot_timeseries(
     name: str,
-    arr: xr.DataArray | xr.Dataset,
+    arr: Union[xr.DataArray, xr.Dataset],
     array_categ: dict[str, str],
     plot_kw: dict[str, Any],
     opts_kw: dict[str, Any],
@@ -250,7 +250,7 @@ def _plot_timeseries(
 
 
 def timeseries(
-    data: dict[str, Any] | xr.DataArray | xr.Dataset,
+    data: Union[dict[str, Any], xr.DataArray, xr.Dataset],
     use_attrs: dict[str, Any] | None = None,
     plot_kw: dict[str, Any] | None = None,
     opts_kw: dict[str, Any] | None = None,

diff --git a/src/figanos/hvplot/utils.py b/src/figanos/hvplot/utils.py
@@ -377,7 +377,7 @@ def x_timeseries(data, plot_kw) -> None:
 
 def set_plot_attrs_hv(
     use_attrs: dict[str, Any],
-    xr_obj: xr.DataArray | xr.Dataset,
+    xr_obj: Union[xr.DataArray, xr.Dataset],
     plot_kw: dict[str, Any],
     wrap_kw: dict[str, Any] | None = None,
 ) -> [dict, dict]:
@@ -522,9 +522,9 @@ def plot_coords_hook(plot, element, text, loc, bgc) -> None:
 
 
 def plot_coords(
-    xr_obj: xr.DataArray | xr.Dataset,
-    loc: str | tuple[float, float] | int,
-    param: str | None = None,
+    xr_obj: Union[xr.DataArray, xr.Dataset],
+    loc: Union[str, tuple[float, float], int],
+    param: Union[str, None] = None,
     backgroundalpha: float = 1,
 ) -> hv.Text:
     """Plot the coordinates of an xarray object.