updated plots

nismod · Dec 18, 2018 · bfa7f16 · bfa7f16
1 parent 1d8adc7
commit bfa7f16
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/energy_demand/plotting/basic_plot_functions.py b/energy_demand/plotting/basic_plot_functions.py
@@ -2,8 +2,7 @@
 """
 import math
 import numpy as np
-import matplotlib.pyplot as plt
-import pylab
+
 from scipy.interpolate import interp1d
 from scipy.interpolate import spline
 
@@ -13,7 +12,7 @@
 def export_legend(legend, filename="legend.png"):
     """Export legend as seperate file
     """
-    fig  = legend.figure
+    fig = legend.figure
     fig.canvas.draw()
     bbox = legend.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
     fig.savefig(filename, dpi="figure", bbox_inches=bbox)