do not plot custom interventions
@@ -144,6 +144,7 @@ def plot_data(sim, ax, key, scatter_args, color=None):
def plot_interventions(sim, ax):
''' Add interventions to the plot '''
for intervention in sim['interventions']:
+ if hasattr(intervention, 'plot'): # Don't plot e.g. functions
intervention.plot(sim, ax)
return
