From ab8656e005cc2d1d018414ab5770c698d92fc321 Mon Sep 17 00:00:00 2001 From: jervdrdo Date: Sun, 12 Dec 2021 13:14:13 +0100 Subject: [PATCH] :bug: minor bug fix in make_robust --- tsflex/features/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tsflex/features/utils.py b/tsflex/features/utils.py index c08dd3fe..a762dc5a 100644 --- a/tsflex/features/utils.py +++ b/tsflex/features/utils.py @@ -120,7 +120,7 @@ def wrap_func(*series: Union[np.ndarray, pd.Series], **kwargs) -> FuncWrapper: if not passthrough_nans: series = [s[~np.isnan(s)] for s in series] if any([len(s) < min_nb_samples for s in series]): - if not isinstance(output_names, list): + if not isinstance(output_names, list) or len(output_names) == 1: return error_val return tuple([error_val] * len(output_names)) return func(*series, **kwargs)