Skip to content

Commit db81c4c

Browse files
fix: extract-lam (#217)
1 parent 5fde9d6 commit db81c4c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/anemoi/inference/outputs/extract_lam.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def __init__(
6363
if "lam_1/cutout_mask" in self.checkpoint.supporting_arrays:
6464
raise NotImplementedError("Only lam_0 is supported")
6565

66-
mask = self.checkpoint.load_supporting_array(f"{lam}/cutout_mask")
66+
mask = self.checkpoint.load_supporting_array(f"0/{lam}/cutout_mask")
6767
assert len(mask) == np.sum(mask)
6868
points = slice(None, np.sum(mask))
6969

0 commit comments

Comments
 (0)