add error handling

genn-team · Mar 7, 2023 · fd7acef · fd7acef
1 parent 0053fcf
commit fd7acef
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ml_genn/ml_genn/callbacks/var_recorder.py b/ml_genn/ml_genn/callbacks/var_recorder.py
@@ -25,6 +25,10 @@ def __init__(self, pop: PopulationType, var: Optional[str], key=None,
             self._var = get_genn_var_name(self._pop.neuron, var)
         elif genn_var is not None:
             self._var = genn_var
+        else:
+            raise RuntimeError("SpikeRecorder callback requires a "
+                               "variable to be specified, either "
+                               "via 'var' or 'genn_var' argument")
 
         # Stash key
         self.key = key