Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Oct 21, 2024
1 parent 695cb5a commit 4fc1f49
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions optimas/generators/ax/service/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,9 @@ def __init__(
model_save_period: Optional[int] = 5,
model_history_dir: Optional[str] = "model_history",
) -> None:
custom_trial_parameters = [ TrialParameter("trial_index", "ax_trial_index", dtype=int)]
custom_trial_parameters = [
TrialParameter("trial_index", "ax_trial_index", dtype=int)
]
super().__init__(
varying_parameters=varying_parameters,
objectives=objectives,
Expand Down Expand Up @@ -170,7 +172,7 @@ def tell(self, results: List[dict]) -> None:
objectives=self._objectives,
analyzed_parameters=self._analyzed_parameters,
custom_parameters=self._custom_trial_parameters,
)
)
try:
trial_id = trial.ax_trial_id
ax_trial = self._ax_client.get_trial(trial_id)
Expand Down

0 comments on commit 4fc1f49

Please sign in to comment.