diff --git a/aion/codecs/preprocessing/spectrum.py b/aion/codecs/preprocessing/spectrum.py index 78429f2..8ee00b2 100644 --- a/aion/codecs/preprocessing/spectrum.py +++ b/aion/codecs/preprocessing/spectrum.py @@ -9,7 +9,7 @@ def pad_spectrum(x: Spectrum) -> Spectrum: Note: Each of the sequence attributes (flux, ivar, mask, wavelength) should be 2D tensors. """ - padding_values = {"lambda": 99999, "mask": True, "ivar": 0} + padding_values = {"wavelength": 99999, "mask": True, "ivar": 0} for k in ["flux", "ivar", "mask", "wavelength"]: setattr(