Add back vectorized support, fix types

mpiannucci · Sep 5, 2024 · a511d43 · a511d43
1 parent 6f3b6b4
commit a511d43
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/gribberish/gribberish_backend.py b/python/gribberish/gribberish_backend.py
@@ -96,12 +96,12 @@ def __init__(
         self.vindex = self
 
     def __getitem__(
-        self, key: xr.core.indexing.ExplicitIndexer | xr.core.indexing.OuterIndexer
+        self, key: xr.core.indexing.ExplicitIndexer
     ) -> np.typing.ArrayLike:
         return indexing.explicit_indexing_adapter(
             key,
             self.shape,
-            indexing.IndexingSupport.OUTER,
+            indexing.IndexingSupport.OUTER_1VECTOR,
             self._raw_indexing_method,
         )