tring to fix tests

bmorris3 · Dec 3, 2018 · 0b02d81 · 0b02d81
1 parent 5288ce4
commit 0b02d81
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/arcesetc/plots.py b/arcesetc/plots.py
@@ -1,4 +1,3 @@
-import matplotlib.pyplot as plt
 import numpy as np
 import astropy.units as u
 from .utils import (closest_target, archive, scale_flux,
@@ -87,6 +86,8 @@ def plot_order_counts(sptype, wavelength, V, exp_time=None,
     wave, flux = matrix_row_to_spectrum(matrix, closest_order)
     flux *= scale_flux(archive[target], V)
 
+    import matplotlib.pyplot as plt
+
     fig, ax = plt.subplots()
 
     if exp_time is not None and signal_to_noise is None:
@@ -188,6 +189,8 @@ def plot_order_sn(sptype, wavelength, V, exp_time=None, signal_to_noise=None,
     wave, flux = matrix_row_to_spectrum(matrix, closest_order)
     flux *= scale_flux(archive[target], V)
 
+    import matplotlib.pyplot as plt
+
     fig, ax = plt.subplots()
 
     if exp_time is not None: