serializable data

Author: BryonLewis

batbot/spectrogram/__init__.py

@@ -1735,12 +1735,14 @@ def compute_wrapper(
 
             segments['stft_db'].append(stft_db[:, start + trim_begin : start + trim_end])
             segment_waveplot = waveplot[:, start + trim_begin : start + trim_end]
-            metadata_waveplot = {
-                "waveplot": segment_waveplot,
-            }
+            segments['waveplot'].append(segment_waveplot)
+            # convert to JSON serializable datatype and add to metadata if segment_waves is True
             if segment_waves:
+                segment_waveplot = segment_waveplot.tolist()
+                metadata_waveplot = {
+                    "waveplot": segment_waveplot,
+                }
                 metadata.update(metadata_waveplot)
-            segments['waveplot'].append(segment_waveplot)
             segments['costs'].append(costs[:, trim_begin:trim_end])
             if debug_path:
                 segments['canvas'].append(canvas[:, trim_begin:trim_end])