updating code to pass all checks

aiidateam · Feb 14, 2025 · fb66e7d · fb66e7d
1 parent 71a6017
commit fb66e7d
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
@@ -186,6 +186,8 @@ def get_y(self) -> list[tuple[str, 'ndarray', str]]:
     def get_y_arraynames(self) -> list[str]:
         """Returns the user-provided names of the y-arrays."""
 
-        y_names = getattr(self.base.attributes, 'y_names', None)
-        if y_names is None:
+        try:
+            y_names = self.base.attributes.get('y_names')
+        except (KeyError, AttributeError):
             raise NotExistent('No y names have been set yet!')
+        return y_names