pyright

experimental-design · Dec 6, 2024 · 2d4a850 · 2d4a850
1 parent 028a2c0
commit 2d4a850
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/bofire/strategies/doe_strategy.py b/bofire/strategies/doe_strategy.py
@@ -5,7 +5,10 @@
 
 import bofire.data_models.strategies.api as data_models
 from bofire.data_models.features.api import CategoricalInput, Input
-from bofire.data_models.strategies.doe import AnyDoEOptimalityCriterion
+from bofire.data_models.strategies.doe import (
+    AnyDoEOptimalityCriterion,
+    DoEOptimalityCriterion,
+)
 from bofire.strategies.doe.branch_and_bound import (
     find_local_max_ipopt_BaB,
     find_local_max_ipopt_exhaustive,
@@ -39,7 +42,7 @@ def __init__(
 
     @property
     def formula(self):
-        if isinstance(self.data_model.criterion, AnyDoEOptimalityCriterion):
+        if isinstance(self.data_model.criterion, DoEOptimalityCriterion):
             return get_formula_from_string(
                 self.data_model.criterion.formula, self.data_model.domain
             )