Skip to content

Commit

Permalink
add supporting_arrays to checkpoints
Browse files Browse the repository at this point in the history
  • Loading branch information
b8raoult committed Oct 26, 2024
1 parent ee5d47d commit 736a758
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/anemoi/utils/checkpoints.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def load_metadata(path: str, *, supporting_arrays=False, name: str = DEFAULT_NAM
import numpy as np

supporting_arrays = {}
for key, entry in metadata.get("supporting_arrays", {}):
for key, entry in metadata.get("supporting_arrays", {}).items():
supporting_arrays[key] = np.frombuffer(
f.read(entry["path"]),
dtype=entry["dtype"],
Expand Down

0 comments on commit 736a758

Please sign in to comment.