Skip to content

Commit

Permalink
cutout must return ints for shape. not np.int64
Browse files Browse the repository at this point in the history
  • Loading branch information
floriankrb committed Nov 20, 2024
1 parent 0cb8f1c commit 6422e30
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/anemoi/datasets/data/grids.py
Original file line number Diff line number Diff line change
Expand Up @@ -324,7 +324,8 @@ def shape(self):
"""
shapes = [np.sum(mask) for mask in self.masks]
global_shape = np.sum(self.global_mask)
return tuple(self.lams[0].shape[:-1] + (sum(shapes) + global_shape,))
total_shape = sum(shapes) + global_shape
return tuple(self.lams[0].shape[:-1] + (int(total_shape),))

def check_same_resolution(self, d1, d2):
# Turned off because we are combining different resolutions
Expand Down

0 comments on commit 6422e30

Please sign in to comment.