Restore peak_waveform_index in units

AllenInstitute · Sep 26, 2024 · 2786002 · 2786002
1 parent 610f2f9
commit 2786002
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/npc_sessions/sessions.py b/src/npc_sessions/sessions.py
@@ -1560,8 +1560,7 @@ def units(self) -> pynwb.misc.Units:
                         f"{group_query} & channel == {row['peak_channel']}"
                     ).index.item()
                 ),
-                # TODO incorrect for some units: add back when annotations are correct
-                # peak_waveform_index=row["electrodes"].index(peak_electrode),
+                peak_waveform_index=row["electrodes"].index(peak_electrode),
                 obs_intervals=self.get_obs_intervals(row["electrode_group_name"]),
             )
         if "waveform_mean" in units: