Issue ecmwf#470, by expanding the selection functionality to include …

…NumPy arrays.
alioacar · Oct 3, 2024 · f952a24 · f952a24
1 parent e07af4f
commit f952a24
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/earthkit/data/core/select.py b/src/earthkit/data/core/select.py
@@ -15,6 +15,7 @@
 
 def normalize_selection(*args, **kwargs):
     from earthkit.data.arguments.transformers import ALL
+    from earthkit.data.utils.array import numpy_backend
 
     _kwargs = {}
     for a in args:
@@ -34,6 +35,7 @@ def normalize_selection(*args, **kwargs):
             or callable(v)
             or isinstance(v, (list, tuple, set, slice))
             or isinstance(v, (str, int, float, datetime.datetime))
+            or numpy_backend()._is_native_array(v)
         ), f"Unsupported type: {type(v)} for key {k}"
     return _kwargs