Skip to content

Commit

Permalink
Add time dim if needed
Browse files Browse the repository at this point in the history
  • Loading branch information
ecomodeller committed Jan 24, 2025
1 parent 0f432d7 commit 56a2e79
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions mikeio/dfs/_dfs.py
Original file line number Diff line number Diff line change
Expand Up @@ -394,13 +394,12 @@ def read(
# TODO this is not used, since Dfs3 has a separate .read method
shape = (nt, self.nz, self.ny, self.nx) # type: ignore

spdims = self.geometry.default_dims
dims = ["time"]
dims.extend(spdims)
dims = self.geometry.default_dims

if single_time_selected and not keepdims:
shape = shape[1:]
dims = dims[1:]
else:
dims = ["time"] + list(dims)

data_list: list[np.ndarray] = [
np.ndarray(shape=shape, dtype=dtype) for _ in range(n_items)
Expand All @@ -421,10 +420,7 @@ def read(
d = d.reshape(self.ny, self.nx) # type: ignore

if single_time_selected:
if keepdims:
data_list[item] = np.atleast_2d(d)
else:
data_list[item] = d
data_list[item] = np.atleast_2d(d) if keepdims else d
else:
data_list[item][i] = d

Expand Down

0 comments on commit 56a2e79

Please sign in to comment.