assert dtype in i, f

SpikeInterface · Jul 18, 2022 · 065e32b · 065e32b
1 parent eb54e8c
commit 065e32b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/MEArec/generators/recordinggenerator.py b/MEArec/generators/recordinggenerator.py
@@ -316,6 +316,9 @@ def generate_recordings(self, tmp_mode=None, tmp_folder=None, n_jobs=0, template
         else:
             params['recordings']['dtype'] = rec_params['dtype']
         dtype = params['recordings']['dtype']
+
+        assert np.dtype(dtype).kind in ("i", "f"), "Only integers and float dtypes are supported"
+
         params['recordings']['adc_bit_depth'] = rec_params.get('adc_bit_depth', None)
         adc_bit_depth = params['recordings']['adc_bit_depth']
         params['recordings']['lsb'] = rec_params.get('lsb', None)