fix(tensorstore): Correctly determine size of nulls

devsjc · devsjc · commit daea7c0a8b88 · 2025-04-02T14:42:14.000+01:00
diff --git a/src/nwp_consumer/internal/entities/tensorstore.py b/src/nwp_consumer/internal/entities/tensorstore.py
@@ -335,7 +335,7 @@ def _calc_null_percentage(data: np.typing.NDArray[np.float32]) -> float:
                     "to coordinate values in the dataset",
                 )
                 return 1.0
-            return float(nulls.sum() / len(nulls))
+            return float(nulls.sum() / np.prod(nulls.shape))
 
         if "latitude" in store_da.dims:
             spatial_dims: list[str] = ["latitude", "longitude"]

Original file line number	Diff line number	Diff line change
`@@ -335,7 +335,7 @@ def _calc_null_percentage(data: np.typing.NDArray[np.float32]) -> float:`
`335`	`335`	`"to coordinate values in the dataset",`
`336`	`336`	`)`
`337`	`337`	`return 1.0`
`338`		`- return float(nulls.sum() / len(nulls))`
	`338`	`+ return float(nulls.sum() / np.prod(nulls.shape))`
`339`	`339`
`340`	`340`	`if "latitude" in store_da.dims:`
`341`	`341`	`spatial_dims: list[str] = ["latitude", "longitude"]`