diff --git a/python/digital_rf/digital_rf_hdf5.py b/python/digital_rf/digital_rf_hdf5.py index 4b57f1c..a00d3a7 100644 --- a/python/digital_rf/digital_rf_hdf5.py +++ b/python/digital_rf/digital_rf_hdf5.py @@ -1491,8 +1491,8 @@ def read_vector_raw( raise IOError(errstr % (start_sample, vector_length, channel_name)) key, z = data_dict.popitem() - # always return 1-D if possible - z = z.squeeze() + # always return 1-D if possible, need atleast_1d in case 1 sample was requested + z = np.atleast_1d(z.squeeze()) if len(z) != vector_length: errstr = "Requested %i samples, but got %i"