Fix spike_amplitudes checks

Author: bjhardcastle

src/npc_ephys/spikeinterface.py

@@ -462,7 +462,7 @@ def spike_amplitudes(self, probe: str) -> tuple[npt.NDArray[np.floating], ...]:
         spike_amplitudes_by_unit: list[npt.NDArray[np.floating]] = []
         for index in sorted(np.unique(unit_indexes)):
             spike_amplitudes_by_unit.append(
-                spike_amplitudes[np.where(unit_indexes == index)[0]]
+                spike_amplitudes[unit_indexes == index]
             )
         return tuple(spike_amplitudes_by_unit)
 

src/npc_ephys/units.py

@@ -225,7 +225,7 @@ def _device_helper(
     ), "Mismatch between rows in spike_times and metrics.csv"
     df_device_metrics["spike_times"] = units_x_spike_times
     df_device_metrics["spike_amplitudes"] = spike_interface_data.spike_amplitudes(electrode_group_name)
-    assert all(len(df_device_metrics["spike_amplitudes"][i]) == len(spike_times) for i, spike_times in enumerate(units_x_spike_times)), "Mismatch between spike_times and spike_amplitudes"
+    assert all(len(df_device_metrics["spike_amplitudes"].iloc[i]) == len(spike_times) for i, spike_times in enumerate(units_x_spike_times)), "Mismatch between spike_times and spike_amplitudes"
 
     return df_device_metrics