Add LFPStimulus base class for lfp stimuli

Author: Jaquier Aurélien Tristan

bluepyopt/ephys/protocols.py

@@ -287,7 +287,7 @@ def instantiate(self, sim=None, cell_model=None):
         """Instantiate"""
 
         for stimulus in self.stimuli:
-            if isinstance(stimulus, LFPySquarePulse):
+            if isinstance(stimulus, LFPStimulus):
                 stimulus.instantiate(lfpy_cell=cell_model.lfpy_cell)
             else:
                 stimulus.instantiate(sim=sim, icell=cell_model.icell)

bluepyopt/ephys/stimuli.py

@@ -32,6 +32,15 @@ class Stimulus(object):
     pass
 
 
+class LFPStimulus(Stimulus):
+
+    """Base class for stimulus supporting LFPy cells."""
+
+    def instantiate():
+        """Run stimulus"""
+        raise NotImplementedError
+
+
 class NrnCurrentPlayStimulus(Stimulus):
 
     """Current stimulus based on current amplitude and time series"""
@@ -330,7 +339,7 @@ def __str__(self):
                 self.location)
 
 
-class LFPySquarePulse(Stimulus):
+class LFPySquarePulse(LFPStimulus):
 
     """Square pulse current clamp injection"""