set fill value for zarr data explicitly

VU-IVM · Sep 18, 2024 · a6c1cd0 · a6c1cd0
1 parent 5d2f0d5
commit a6c1cd0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/honeybees/reporter.py b/honeybees/reporter.py
@@ -165,12 +165,21 @@ def export_value(self, name: str, value: np.ndarray, conf: dict) -> None:
                     compressor = zstd_compressor
                     dtype = value.dtype
                     array_dimensions = ["time", "agents"]
+                if dtype in (float, np.float32, np.float64):
+                    fill_value = np.nan
+                elif dtype in (int, np.int32, np.int64):
+                    fill_value = -1
+                else:
+                    raise ValueError(
+                        f"Value {dtype} of type {type(dtype)} not recognized."
+                    )
                 ds.create_dataset(
                     name,
                     shape=shape,
                     chunks=chunks,
                     dtype=dtype,
                     compressor=compressor,
+                    fill_value=fill_value,
                 )
                 ds[name].attrs["_ARRAY_DIMENSIONS"] = array_dimensions
             index = conf["_time_index"].index(self.model.current_time)