Fix dummy data creation for ctapipe 0.21

cta-observatory · Jul 19, 2024 · 6e544a9 · 6e544a9
1 parent e693145
commit 6e544a9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ctapipe_io_lst/tests/test_cta_r1.py b/src/ctapipe_io_lst/tests/test_cta_r1.py
@@ -91,6 +91,8 @@ def create_shower(rng):
 
 def create_waveform(image, peak_time, num_samples=40, gains=(86, 5), offset=400):
     r1 = waveform_model.get_waveform(image, peak_time, num_samples)
+    if CTAPIPE_0_21:
+        r1 = r1[0]
     return np.array([r1 * gain + offset for gain in gains]).astype(np.uint16)