Silence warning by allowing extra columns

emdgroup · Jul 22, 2024 · b6c56e9 · b6c56e9
1 parent 2bbff23
commit b6c56e9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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()
             )