diff --git a/src/toolbox_scs/detectors/viking.py b/src/toolbox_scs/detectors/viking.py index 476c527e4a80fe886c4a67b59672961abe59c249..ff96db9ece0686d51ba403ac79cb86387e9e9de6 100644 --- a/src/toolbox_scs/detectors/viking.py +++ b/src/toolbox_scs/detectors/viking.py @@ -290,6 +290,7 @@ class Viking: """ if key not in data: return + print('hello') x = data.newt_x spectra = data[key] mask = xr.ones_like(x, dtype=bool) @@ -300,7 +301,8 @@ class Viking: ranges = self.BL_SIGNAL_RANGE for xrange in ranges: mask = mask & ((x < xrange[0]) | (x > xrange[1])) - x_bl = x.where(mask, drop=True) + x_bl = x.where(mask, drop=True).astype(int) + print(x_bl) bl = spectra.sel(newt_x=x_bl) fit = np.polyfit(x_bl, bl.T, self.BL_POLY_DEG) if len(spectra.shape) == 1: