Commit 1c8729de authored by Andrew Quinn's avatar Andrew Quinn
Browse files

Link gesd_args through in detect_artefacts

parent 39fdd8a6
Loading
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -378,13 +378,13 @@ def detect_artefacts(X, axis=None, reject_mode='dim', metric_func=np.std,
        raise ValueError('bad axis')

    if reject_mode == 'dim':
        bad_inds = _find_outliers_in_dims(X, axis=axis, metric_func=metric_func, gesd_args=None)
        bad_inds = _find_outliers_in_dims(X, axis=axis, metric_func=metric_func, gesd_args=gesd_args)

    elif reject_mode == 'segments':
        bad_inds = _find_outliers_in_segments(X, axis=axis,
                                              segment_len=segment_len,
                                              metric_func=metric_func,
                                              gesd_args=None)
                                              gesd_args=gesd_args)

    if ret_mode == 'bad_inds':
        return bad_inds