fix dtype when using xr.where()

src/toolbox_scs/detectors/viking.py

@@ -309,6 +309,7 @@ class Viking:
             return
         x = data.newt_x
         spectra = data[key]
+        #print(x)
         mask = xr.ones_like(x, dtype=bool)
         if len(self.BL_SIGNAL_RANGE) > 0:
             if not hasattr(self.BL_SIGNAL_RANGE[0], '__len__'):
@@ -317,8 +318,11 @@ 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).astype(int)
+        x_bl = x.where(mask, drop=True).astype(x.dtype)
         bl = spectra.sel(newt_x=x_bl)
+        #print(mask)
+        #print(x_bl)
+        #print(bl)
         fit = np.polyfit(x_bl, bl.T, self.BL_POLY_DEG)
         if len(spectra.shape) == 1:
             return spectra - np.polyval(fit, x)