diff --git a/src/gt4py/next/iterator/transforms/global_tmps.py b/src/gt4py/next/iterator/transforms/global_tmps.py index 0573a36bfa..135379b72c 100644 --- a/src/gt4py/next/iterator/transforms/global_tmps.py +++ b/src/gt4py/next/iterator/transforms/global_tmps.py @@ -407,7 +407,7 @@ def _max_domain_sizes_by_location_type(offset_provider: Mapping[str, Any]) -> di ) sizes[provider.neighbor_axis.value] = max( sizes.get(provider.neighbor_axis.value, 0), - provider.table.max(), # type: ignore[attr-defined] # TODO(havogt): improve typing for NDArrayObject + provider.table.max() + 1, # type: ignore[attr-defined] # TODO(havogt): improve typing for NDArrayObject ) return sizes