Fix Connectivity protocol

GridTools · Jan 12, 2023 · 82b5c26 · 82b5c26
1 parent 4d8ab91
commit 82b5c26
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/functional/common.py b/src/functional/common.py
@@ -77,6 +77,7 @@ class Connectivity(Protocol):
     max_neighbors: int
     has_skip_values: bool
     origin_axis: Dimension
+    neighbor_axis: Dimension
 
     def mapped_index(
         self, cur_index: int | np.integer, neigh_index: int | np.integer