From beadae38787c8ca0cced0071b8cb6f4bd2c6b41b Mon Sep 17 00:00:00 2001 From: Chris Holdgraf Date: Tue, 29 Aug 2017 23:14:26 -0400 Subject: [PATCH] mean fill bug --- mne/decoding/receptive_field.py | 2 +- mne/decoding/tests/test_receptive_field.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/mne/decoding/receptive_field.py b/mne/decoding/receptive_field.py index 1f55f977ccb..0407905ebf3 100644 --- a/mne/decoding/receptive_field.py +++ b/mne/decoding/receptive_field.py @@ -411,7 +411,7 @@ def _delay_time_series(X, tmin, tmax, sfreq, newaxis=0, axis=0, fill_value = X.mean(axis=axis, keepdims=True) if epoch_axis is not None: fill_value = np.mean(fill_value, axis=epoch_axis, keepdims=True) - delayed[...] = fill_value + delayed[...] = fill_value for ii, ix_delay in enumerate(delays): take = [slice(None)] * X.ndim put = [slice(None)] * X.ndim diff --git a/mne/decoding/tests/test_receptive_field.py b/mne/decoding/tests/test_receptive_field.py index e1ec45c8e48..1d9da36e4c8 100644 --- a/mne/decoding/tests/test_receptive_field.py +++ b/mne/decoding/tests/test_receptive_field.py @@ -123,6 +123,12 @@ def test_receptive_field(): rf.fit(X, y) assert_array_equal(rf.delays_, np.arange(tmin, tmax + 1)) + # Testing fill mean + rf = ReceptiveField(tmin, tmax, 1, feature_names, fill_mean=True, + estimator=mod) + rf.fit(X, y) + assert_array_equal(rf.delays_, np.arange(tmin, tmax + 1)) + y_pred = rf.predict(X) assert_allclose(y[rf.valid_samples_], y_pred[rf.valid_samples_], atol=1e-2) scores = rf.score(X, y)