diff --git a/tardis/montecarlo/spectrum.py b/tardis/montecarlo/spectrum.py index ad13b1e56ad..857fed5bb1e 100644 --- a/tardis/montecarlo/spectrum.py +++ b/tardis/montecarlo/spectrum.py @@ -103,13 +103,13 @@ def luminosity_to_flux(luminosity, distance): def f_nu_to_f_lambda(self, f_nu): return f_nu * self.frequency / self.wavelength - def plot(self, ax=None, mode="wavelength"): + def plot(self, ax=None, mode="wavelength", **kwargs): if ax is None: from matplotlib.pyplot import gca ax = gca() if mode == "wavelength": - ax.plot(self.wavelength.value, self.luminosity_density_lambda.value) + ax.plot(self.wavelength.value, self.luminosity_density_lambda.value, **kwargs) ax.set_xlabel( "Wavelength [{}]".format(self.wavelength.unit._repr_latex_()) )