Fix pre-commit issues (#77)

nanoribbon/viewers/igor.py

@@ -19,6 +19,11 @@ def __init__(self):
         super().__init__("Missing 'BEGIN' statement of data block")
 
 
+class UnknownParameterError(KeyError):
+    def __init__(self, key):
+        super().__init__(f"Unknown parameter {key}")
+
+
 class Axis:
     """Represents an axis of an IGOR wave"""
 
@@ -174,7 +179,7 @@ def __init__(self, data=None, axes=None, name="1d", **kwargs):
             if key in self.parameters:
                 self.parameters[key] = value
             else:
-                raise KeyError(f"Unknown parameter {key}")
+                raise UnknownParameterError(key)
 
         if axes is None:
             p = self.parameters
@@ -225,7 +230,7 @@ def __init__(self, data=None, axes=None, name=None, **kwargs):
             if key in self.parameters:
                 self.parameters[key] = value
             else:
-                raise KeyError(f"Unknown parameter {key}")
+                raise UnknownParameterError(key)
 
         if axes is None:
             p = self.parameters

nanoribbon/viewers/search.py

@@ -66,6 +66,13 @@ def __init__(self, label, state):
         super().__init__(f"Calculation {label} in state {state}.")
 
 
+class WorkchainExceptedError(RuntimeError):
+    def __init__(self, workchain):
+        super().__init__(
+            f"Workchain {workchain.pk} excepted ({workchain.exit_status})."
+        )
+
+
 class NanoribbonSearchWidget(ipw.VBox):
     STYLE = {"description_width": "120px"}
     LAYOUT = ipw.Layout(width="80%")
@@ -186,9 +193,7 @@ def get_calc_by_label(workcalc, label):
             return calc
 
         if workcalc.exit_status or workcalc.is_excepted:
-            raise RuntimeError(
-                f"Workchain {workcalc.pk} in state {workcalc.exit_status}."
-            )
+            raise WorkchainExceptedError(workcalc)
         # Formula and description.
         structure = workcalc.inputs.structure
         ase_struct = structure.get_ase()