improve exception handling

aiidateam · Feb 14, 2025 · a9e7df5 · a9e7df5
1 parent ccb9d2a
commit a9e7df5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/aiida/orm/nodes/data/array/xy.py b/src/aiida/orm/nodes/data/array/xy.py
@@ -187,6 +187,8 @@ def get_y_arraynames(self) -> list[str]:
         """Returns the user-provided names of the y-arrays."""
 
         try:
-            return self.base.attributes.get('y_names')
+            y_names = self.base.attributes.get('y_names')
+            if y_names is None:
+                raise KeyError
         except (KeyError, AttributeError):
-            raise NotExistent(f'No y names been set yet')
+            raise NotExistent(f'No y names been set yet!')