Add resultant kwarg so intensity and tune test are more flexible (#70)

wright-group · Nov 8, 2019 · 474f1e7 · 474f1e7
1 parent c71ff6a
commit 474f1e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/attune/workup/_intensity.py b/attune/workup/_intensity.py
@@ -14,7 +14,7 @@
 
 
 def _intensity(data, channel_name, tune_points, *, spline=True, **spline_kwargs):
-    data.moment(axis=1, channel=channel_name, moment=1)
+    data.moment(axis=1, channel=channel_name, moment=1, resultant=data.axes[0].shape)
     offsets = data[f"{channel_name}_1_moment_1"].points
 
     if spline:

diff --git a/attune/workup/_tune_test.py b/attune/workup/_tune_test.py
@@ -12,7 +12,7 @@
 
 
 def _offsets(data, channel_name, tune_points, *, spline=True, **spline_kwargs):
-    data.moment(axis=1, channel=channel_name, moment=1)
+    data.moment(axis=1, channel=channel_name, moment=1, resultant=data.axes[0].shape)
     offsets = data[f"{channel_name}_1_moment_1"].points
 
     if spline: