No need to pass is_fidelity

Author: RemiLehe

optimas/generators/ax/service/multi_fidelity.py

@@ -134,7 +134,6 @@ def _create_generation_steps(
     def set_fidelity_param(
         self,
         var_name: str,
-        is_fidelity: bool = True,
         fidelity_target_value: float = None,
     ) -> None:
         """Set a parameter as the fidelity parameter for multi-fidelity optimization.
@@ -143,8 +142,6 @@ def set_fidelity_param(
         ----------
         var_name : str
             Name of the variable to set as fidelity parameter.
-        is_fidelity : bool, optional
-            Whether this parameter is the fidelity parameter. Default True.
         fidelity_target_value : float, optional
             The target fidelity value for optimization.
         """
@@ -159,7 +156,7 @@ def set_fidelity_param(
                 f"Variable '{var_name}' not found in varying parameters"
             )
 
-        var.is_fidelity = is_fidelity
+        var.is_fidelity = True
         if fidelity_target_value is not None:
             var.fidelity_target_value = fidelity_target_value
 

tests/test_ax_generators.py

@@ -498,7 +498,7 @@ def test_ax_multi_fidelity():
     )
 
     gen = AxMultiFidelityGenerator(vocs=vocs)
-    gen.set_fidelity_param("res", is_fidelity=True, fidelity_target_value=8.0)
+    gen.set_fidelity_param("res", fidelity_target_value=8.0)
     ev = FunctionEvaluator(function=eval_func_mf)
     exploration = Exploration(
         generator=gen,
@@ -697,7 +697,7 @@ def test_ax_multi_fidelity_with_history():
     )
 
     gen = AxMultiFidelityGenerator(vocs=vocs)
-    gen.set_fidelity_param("res", is_fidelity=True, fidelity_target_value=8.0)
+    gen.set_fidelity_param("res", fidelity_target_value=8.0)
     ev = FunctionEvaluator(function=eval_func_mf)
     exploration = Exploration(
         generator=gen,