From 737c93723bfae5d188aa5b63e7c4524bd74110d7 Mon Sep 17 00:00:00 2001 From: Martin Fitzner Date: Sat, 7 Sep 2024 20:15:13 +0200 Subject: [PATCH] Add comment --- baybe/recommenders/pure/bayesian/base.py | 2 ++ 1 file changed, 2 insertions(+) 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(