Re-ordered guvectorize and updated unit test

cta-observatory · Nov 2, 2023 · d035cfd · d035cfd
1 parent ba646c7
commit d035cfd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ctapipe/image/extractor.py b/ctapipe/image/extractor.py
@@ -352,8 +352,8 @@ def integration_correction(
 
 @guvectorize(
     [
-        (float64[:], float32[:], float32[:], float32[:]),
         (float32[:], float32[:], float32[:], float32[:]),
+        (float64[:], float32[:], float32[:], float32[:]),
     ],
     "(s)->(),(),()",
     nopython=True,

diff --git a/ctapipe/image/tests/test_extractor.py b/ctapipe/image/tests/test_extractor.py
@@ -163,15 +163,15 @@ def test_fwhm(toymodel):
 
     for i in range(0, len(waveforms)):
         peak_index = np.argmax(waveforms[i])
-        width, width_height, l_ips, r_ips = peak_widths(
+        widths = peak_widths(
             waveforms[i],
             peaks=[
                 peak_index,
             ],
             rel_height=0.5,
         )
 
-        fwhm_scp = np.append(fwhm_scp, width)
+        fwhm_scp = np.append(fwhm_scp, widths[0])
 
     fwhm, _, _ = time_parameters(
         waveforms,