diff --git a/backtrader/cerebro.py b/backtrader/cerebro.py index 9b18e7775..9a87d6f15 100644 --- a/backtrader/cerebro.py +++ b/backtrader/cerebro.py @@ -938,8 +938,15 @@ def getbroker(self): broker = property(getbroker, setbroker) + def get_figures(self, plotter=None, numfigs=1, iplot=True, start=None, end=None, + width=16, height=9, dpi=300, tight=True, use=None, **kwargs): + + return self.plot(plotter=None, numfigs=1, iplot=True, start=None, end=None, + width=16, height=9, dpi=300, tight=True, use=None, show_plots=False, + **kwargs) + def plot(self, plotter=None, numfigs=1, iplot=True, start=None, end=None, - width=16, height=9, dpi=300, tight=True, use=None, + width=16, height=9, dpi=300, tight=True, use=None, show_plots=True, **kwargs): ''' Plots the strategies inside cerebro @@ -998,7 +1005,8 @@ def plot(self, plotter=None, numfigs=1, iplot=True, start=None, end=None, figs.append(rfig) - plotter.show() + if show_plots: + plotter.show() return figs