Skip to content

Commit 9b60771

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent cf1211d commit 9b60771

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

optimas/generators/ax/service/multi_fidelity.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -132,10 +132,13 @@ def _create_generation_steps(
132132
return steps
133133

134134
def set_fidelity_param(
135-
self, var_name: str, is_fidelity: bool = True, fidelity_target_value: float = None
135+
self,
136+
var_name: str,
137+
is_fidelity: bool = True,
138+
fidelity_target_value: float = None,
136139
) -> None:
137140
"""Set a parameter as the fidelity parameter for multi-fidelity optimization.
138-
141+
139142
Parameters
140143
----------
141144
var_name : str
@@ -150,13 +153,15 @@ def set_fidelity_param(
150153
if vp.name == var_name:
151154
var = vp
152155
break
153-
156+
154157
if var is None:
155-
raise ValueError(f"Variable '{var_name}' not found in varying parameters")
156-
158+
raise ValueError(
159+
f"Variable '{var_name}' not found in varying parameters"
160+
)
161+
157162
var.is_fidelity = is_fidelity
158163
if fidelity_target_value is not None:
159164
var.fidelity_target_value = fidelity_target_value
160-
165+
161166
# Update the Ax client
162167
self._update_parameter(var)

0 commit comments

Comments
 (0)