fix NWB export - null waveforms - back to version 0.1.0b1

Author: Thinh Nguyen

CHANGELOG.md

@@ -3,7 +3,7 @@
 Observes [Semantic Versioning](https://semver.org/spec/v2.0.0.html) standard and [Keep a Changelog](https://keepachangelog.com/en/1.0.0/) convention.
 
 
-## [0.1.0b2] - 2021-11-29
+## [0.1.0b1] - 2021-11-29
 ### Added
 + NWB export
 + Processing with Kilosort and pyKilosort for Open Ephys and SpikeGLX

element_array_ephys/export/nwb.py

@@ -100,16 +100,17 @@ def curated_clusterings_to_nwb(curated_clustering_keys, nwbfile, ephys_module=No
         # ---- Units ----
         unit_query = clustering_query @ ephys.CuratedClustering.Unit
         for unit in unit_query.fetch(as_dict=True):
+            waveform_mean = waveform_std = np.full(1, np.nan)
             if ephys.WaveformSet & curated_clustering_key & {'unit': unit['unit']}:
                 waveform_mean = (ephys.WaveformSet.PeakWaveform
                                  & curated_clustering_key
                                  & {'unit': unit['unit']}).fetch1('peak_electrode_waveform')
                 # compute waveform's STD across spikes from the peak electrode
-                waveform_std = np.std((ephys.WaveformSet.Waveform & curated_clustering_key
+                electrode_waveforms = (ephys.WaveformSet.Waveform & curated_clustering_key
                                        & {'unit': unit['unit']}
-                                       & {'electrode': unit['electrode']}).fetch1('waveforms'), axis=0)
-            else:
-                waveform_mean = waveform_std = np.full(1, np.nan)
+                                       & {'electrode': unit['electrode']}).fetch1('waveforms')
+                if electrode_waveforms is not None:
+                    waveform_std = np.std(electrode_waveforms, axis=0)
 
             nwbfile.add_unit(id=unit['unit'],
                              electrodes=np.where(nwbfile.electrodes.id.data == int(unit['electrode']))[0],

element_array_ephys/version.py

@@ -1,2 +1,2 @@
 """Package metadata."""
-__version__ = '0.1.0b2'
+__version__ = '0.1.0b1'