Drop unnecessary validator

emdgroup · Sep 6, 2024 · 3e31b7d · 3e31b7d
1 parent 3947528
commit 3e31b7d
Showing 1 changed file with 0 additions and 9 deletions.
diff --git a/baybe/surrogates/bandits/base.py b/baybe/surrogates/bandits/base.py
@@ -9,7 +9,6 @@
 from baybe.objectives.base import Objective
 from baybe.objectives.single import SingleTargetObjective
 from baybe.priors.base import Prior
-from baybe.priors.basic import BetaPrior
 from baybe.searchspace.core import SearchSpace
 from baybe.surrogates.bandits.beta_bernoulli import (
     BetaBernoulliMultiArmedBanditSurrogate,
@@ -45,14 +44,6 @@ class MultiArmedBanditSurrogate:
     _bandit_model: Surrogate | None = field(init=False, default=None, eq=False)
     """The specific bandit model to which is being dispatched."""
 
-    @prior.validator
-    def _validate_prior(self, _, value):
-        if not isinstance(value, BetaPrior):
-            raise NotImplementedError(
-                f"Currently, only priors of type '{BetaPrior.__name__}' "
-                f"are supported."
-            )
-
     def fit(
         self, searchspace: SearchSpace, objective: Objective, measurements: DataFrame
     ) -> None: