Add units to amplitude annotation of SpikeTrains

Closes #281
jpgill86 · Dec 21, 2020 · 8b1a87f · 8b1a87f
1 parent d6e6076
commit 8b1a87f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neurotic/datasets/data.py b/neurotic/datasets/data.py
@@ -620,7 +620,7 @@ def _detect_spikes(sig, discriminator, epochs):
     st = neo.SpikeTrain(
         name = discriminator['name'],
         channels = [discriminator['channel']],  # custom annotation
-        amplitude = discriminator['amplitude'], # custom annotation
+        amplitude = pq.Quantity(discriminator['amplitude'], discriminator['units']), # custom annotation
         times = spikes_between_min_and_max * pq.s,
         t_start = sig.t_start,
         t_stop  = sig.t_stop,