Fix handling of pending experiments in naive recommender

emdgroup · Jan 23, 2025 · 4034b83 · 4034b83
1 parent 4dc4f2f
commit 4034b83
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/baybe/recommenders/naive.py b/baybe/recommenders/naive.py
@@ -58,8 +58,9 @@ def recommend(
     ) -> pd.DataFrame:
         from baybe.acquisition.partial import PartialAcquisitionFunction
 
-        if (not isinstance(self.disc_recommender, BayesianRecommender)) and (
-            not isinstance(self.disc_recommender, NonPredictiveRecommender)
+        disc_is_bayesian = isinstance(self.disc_recommender, BayesianRecommender)
+        if not disc_is_bayesian and not isinstance(
+            self.disc_recommender, NonPredictiveRecommender
         ):
             raise NotImplementedError(
                 """The discrete recommender should be either a Bayesian or a
@@ -79,7 +80,7 @@ def recommend(
                 searchspace=searchspace,
                 objective=objective,
                 measurements=measurements,
-                pending_experiments=pending_experiments,
+                pending_experiments=pending_experiments if disc_is_bayesian else None,
             )
 
         # We are in a hybrid setting now