From b6c56e9c2420ae274b4f8551d78cf25496452626 Mon Sep 17 00:00:00 2001 From: AdrianSosic Date: Mon, 22 Jul 2024 10:10:34 +0200 Subject: [PATCH] Silence warning by allowing extra columns --- baybe/surrogates/base.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/baybe/surrogates/base.py b/baybe/surrogates/base.py index c4944fb7f..56f23e5bd 100644 --- a/baybe/surrogates/base.py +++ b/baybe/surrogates/base.py @@ -235,7 +235,9 @@ def fit( def transform_inputs(df: pd.DataFrame, /) -> pd.DataFrame: """Fitted input transformation pipeline.""" - out = input_scaler.transform(searchspace.transform(df, allow_missing=True)) + out = input_scaler.transform( + searchspace.transform(df, allow_extra=True, allow_missing=True) + ) return pd.DataFrame( out, index=df.index, columns=input_scaler.get_feature_names_out() )