check key in annotations

h-mayorquin · Feb 11, 2024 · a1a5846 · a1a5846
1 parent c96ec93
commit a1a5846
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/neo/core/baseneo.py b/neo/core/baseneo.py
@@ -45,7 +45,8 @@ def _check_annotations(value):
         if not issubclass(value.dtype.type, ALLOWED_ANNOTATION_TYPES):
             raise ValueError(f"Invalid annotation. NumPy arrays with dtype {value.dtype.type}" f"are not allowed")
     elif isinstance(value, dict):
-        for element in value.values():
+        for key, element in value.items():
+            _check_annotations(key)
             _check_annotations(element)
     elif isinstance(value, (list, tuple)):
         for element in value: