Fix mypy failure

pymc-devs · Oct 2, 2023 · a3ec9a5 · a3ec9a5
1 parent 55e8fe9
commit a3ec9a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc/data.py b/pymc/data.py
@@ -234,7 +234,7 @@ def determine_coords(
             for dim in dims:
                 dim_name = dim
                 # str is applied because dim entries may be None
-                coords[str(dim_name)] = value[dim].to_numpy()
+                coords[str(dim_name)] = cast(xr.DataArray, value[dim]).to_numpy()
 
     if isinstance(value, np.ndarray) and dims is not None:
         if len(dims) != value.ndim: