diff --git a/baybe/recommenders/pure/bayesian/base.py b/baybe/recommenders/pure/bayesian/base.py index 7f85d5dfd..aeb39bfab 100644 --- a/baybe/recommenders/pure/bayesian/base.py +++ b/baybe/recommenders/pure/bayesian/base.py @@ -65,7 +65,9 @@ def get_surrogate( measurements: pd.DataFrame, ) -> SurrogateProtocol: """Get the trained surrogate model.""" + # This fit applies internal caching and does not necessarily involve computation self._surrogate_model.fit(searchspace, objective, measurements) + return self._surrogate_model def _setup_botorch_acqf(