Move mpl imports to function levels (closes #6)

mrakitin · Jan 15, 2018 · 14786d0 · 14786d0
1 parent bd87d23
commit 14786d0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/bnlcrl/plot_delta.py b/bnlcrl/plot_delta.py
@@ -1,9 +1,8 @@
-from matplotlib import pyplot as plt
-
 from bnlcrl.delta_finder import DeltaFinder
 
 if __name__ == '__main__':
     import numpy as np
+    from matplotlib import pyplot as plt
 
     step = 1000
     l = np.arange(30, 30000 + 1, step)

diff --git a/bnlcrl/visualize.py b/bnlcrl/visualize.py
@@ -1,11 +1,11 @@
 from io import StringIO  # StringIO behaves like a file object
 
 import numpy as np
-from matplotlib import pyplot as plt
 
 
 def plot_data(df, elements, property, thickness, e_min, e_max, n_points, file_name='data', x_label=None,
               figsize=(10, 6), show_plot=False):
+    from matplotlib import pyplot as plt
     thickness = ', thickness={} [$\mathrm{{\mu}}$m]'.format(thickness) if property == 'transmission' else ''
 
     fig = plt.figure(figsize=figsize)