Merge pull request #64 from wright-group/all_nan_slice

Add case for all nan slice in setpoint
wright-group · Aug 7, 2019 · 1ec53d9 · 1ec53d9
2 parents 0ef5e2d + 0368d42
commit 1ec53d9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/attune/workup/_setpoint.py b/attune/workup/_setpoint.py
@@ -21,7 +21,9 @@ def _setpoint(data, channel_name, tune_points, *, spline=True, **spline_kwargs):
         xi = c.axes[0].points
         yi = c[channel_name].points
         xi, yi = wt.kit.remove_nans_1D(xi, yi)
-        if np.nanmin(yi) <= 0 <= np.nanmax(yi):
+        if np.all(np.isnan(yi)):
+            offsets.append(np.nan)
+        elif np.nanmin(yi) <= 0 <= np.nanmax(yi):
             p = np.polynomial.Polynomial.fit(yi, xi, 2)
             offsets.append(p(0))
         else: