From 75c2753b98ee0136216a404d901e8c1132483691 Mon Sep 17 00:00:00 2001 From: Victor Grau Serrat Date: Fri, 22 Sep 2017 11:59:57 -0600 Subject: [PATCH] matplotlib imports inside init live_graph_clock --- catalyst/exchange/live_graph_clock.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/catalyst/exchange/live_graph_clock.py b/catalyst/exchange/live_graph_clock.py index 60e7503b..e84ee719 100644 --- a/catalyst/exchange/live_graph_clock.py +++ b/catalyst/exchange/live_graph_clock.py @@ -12,7 +12,6 @@ # limitations under the License. from datetime import timedelta -import matplotlib.dates as mdates import pandas as pd from catalyst.gens.sim_engine import ( BAR, @@ -24,8 +23,6 @@ from matplotlib import style log = Logger('LiveGraphClock') -fmt = mdates.DateFormatter('%Y-%m-%d %H:%M') - class LiveGraphClock(object): """Realtime clock for live trading. @@ -55,11 +52,17 @@ class LiveGraphClock(object): def __init__(self, sessions, context, time_skew=pd.Timedelta('0s')): + import matplotlib.dates as mdates + from matplotlib import pyplot as plt + from matplotlib import style + + self.sessions = sessions self.time_skew = time_skew self._last_emit = None self._before_trading_start_bar_yielded = True self.context = context + self.fmt = mdates.DateFormatter('%Y-%m-%d %H:%M') style.use('dark_background') @@ -97,7 +100,7 @@ class LiveGraphClock(object): :return: """ ax.xaxis.set_major_locator(mdates.DayLocator(interval=1)) - ax.xaxis.set_major_formatter(fmt) + ax.xaxis.set_major_formatter(self.fmt) locator = mdates.HourLocator(interval=4) locator.MAXTICKS = 5000