#131 Fix inferred type of abs(DataArray)

deshima-dev · Nov 12, 2023 · 8a9a8c1 · 8a9a8c1
1 parent 8fc5cda
commit 8a9a8c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/decode/utils.py b/decode/utils.py
@@ -2,7 +2,7 @@
 
 
 # dependencies
-from typing import Any, Optional
+from typing import Any, Optional, cast
 import numpy as np
 import xarray as xr
 from xarray.core.types import Dims
@@ -37,4 +37,4 @@ def median(da: xr.DataArray) -> xr.DataArray:
             **kwargs,
         )
 
-    return median(np.abs(da - median(da)))
+    return median(cast(xr.DataArray, np.abs(da - median(da))))