diff --git a/pyaldata/tools.py b/pyaldata/tools.py index 6847703..104bf55 100644 --- a/pyaldata/tools.py +++ b/pyaldata/tools.py @@ -86,7 +86,7 @@ def rebin_array(arr, red_fun): arr = arr[:T, :] arr = arr.reshape(int(T / n_bins), n_bins, N) - return red_fun(arr, axis=1).squeeze() + return red_fun(arr, axis=1).reshape(-1, N) # start with spike fields for col in spike_fields: