added option for keyword arguments in spectrum plot function (#1477)

tardis-sn · Mar 24, 2021 · f545ede · f545ede
1 parent 110d8c9
commit f545ede
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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_())
             )