Add max_suggest_attempts as an attribute

Signed-off-by: Fabrice Normandin <fabrice.normandin@gmail.com>
Epistimio · Apr 14, 2022 · 872f094 · 872f094
1 parent 619d6d6
commit 872f094
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/orion/core/worker/primary_algo.py b/src/orion/core/worker/primary_algo.py
@@ -73,6 +73,7 @@ def __init__(self, space: Space, algorithm: AlgoType):
             original_registry=self.registry,
             transformed_registry=self.algorithm.registry,
         )
+        self.max_suggest_attempts = 100
 
     @property
     def original_space(self) -> Space:
@@ -139,9 +140,8 @@ def suggest(self, num: int) -> list[Trial] | None:
         """
 
         trials: list[Trial] = []
-        max_suggest_attempts = 100
 
-        for suggest_attempt in range(1, max_suggest_attempts + 1):
+        for suggest_attempt in range(1, self.max_suggest_attempts + 1):
             transformed_trials: list[Trial] | None = self.algorithm.suggest(num)
             transformed_trials = transformed_trials or []