add comment explaining types

alan-turing-institute · Jan 15, 2025 · 191dfd7 · 191dfd7
1 parent 2234cc2
commit 191dfd7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deepsensor/data/utils.py b/deepsensor/data/utils.py
@@ -91,6 +91,8 @@ def compute_xarray_data_resolution(ds: Union[xr.DataArray, xr.Dataset]) -> float
     """
     x1_res = np.abs(np.mean(np.diff(ds["x1"])))
     x2_res = np.abs(np.mean(np.diff(ds["x2"])))
+
+    # ensure float type, since numpy 2, np.mean returns a numpy float32
     data_resolution = float(np.min([x1_res, x2_res]))
     return data_resolution