Skip to content

Commit

Permalink
rename
Browse files Browse the repository at this point in the history
  • Loading branch information
bpkroth committed Oct 4, 2024
1 parent b531507 commit 0a243cf
Show file tree
Hide file tree
Showing 8 changed files with 15 additions and 15 deletions.
2 changes: 1 addition & 1 deletion mlos_bench/mlos_bench/config/optimizers/manual_opt.jsonc
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,6 @@

"config": {
"max_cycles": 30,
"cycle_tunable_values": []
"tunable_values_cycle": []
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -232,7 +232,7 @@
"type": "integer",
"minimum": 1
},
"cycle_tunable_values": {
"tunable_values_cycle": {
"description": "The tunable values to cycle through.",
"type": "array",
"items": {
Expand Down
16 changes: 8 additions & 8 deletions mlos_bench/mlos_bench/optimizers/manual_optimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,23 +26,23 @@ def __init__(
service: Optional[Service] = None,
):
super().__init__(tunables, config, global_config, service)
self._cycle_tunable_values: List[Dict[str, TunableValue]] = config.get(
"cycle_tunable_values", []
self._tunable_values_cycle: List[Dict[str, TunableValue]] = config.get(
"tunable_values_cycle", []
)
if len(self._cycle_tunable_values) == 0:
_LOG.warning("No cycle_tunable_values provided, using default values.")
self._cycle_tunable_values = [tunables.get_param_values()]
if len(self._tunable_values_cycle) == 0:
_LOG.warning("No tunable_values_cycle provided, using default values.")
self._tunable_values_cycle = [tunables.get_param_values()]
max_cycles = int(config.get("max_cycles", 1))
self._max_suggestions = min(
self._max_suggestions,
max_cycles * len(self._cycle_tunable_values),
max_cycles * len(self._tunable_values_cycle),
)

def suggest(self) -> TunableGroups:
"""Always produce the same sequence of explicit suggestions, in a cycle."""
tunables = super().suggest()
cycle_index = (self._iter - 1) % len(self._cycle_tunable_values)
tunables.assign(self._cycle_tunable_values[cycle_index])
cycle_index = (self._iter - 1) % len(self._tunable_values_cycle)
tunables.assign(self._tunable_values_cycle[cycle_index])
_LOG.info("Iteration %d :: Suggest: %s", self._iter, tunables)
return tunables

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,6 @@
"config": {
// max_cycles should be at least 1
"max_cycles": 0,
"cycle_tunable_values": []
"tunable_values_cycle": []
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
"class": "mlos_bench.optimizers.ManualOptimizer",

"config": {
"cycle_tunable_values": [],
"tunable_values_cycle": [],
"extra_param": "should not be here"
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
"seed": 12345,
"start_with_defaults": false,
"max_cycles": 10,
"cycle_tunable_values": [
"tunable_values_cycle": [
{
"param1": "value1",
"param2": 1,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,6 @@

"class": "mlos_bench.optimizers.ManualOptimizer",
"config": {
"cycle_tunable_values": []
"tunable_values_cycle": []
}
}
2 changes: 1 addition & 1 deletion mlos_bench/mlos_bench/tests/optimizers/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,6 @@ def manual_opt(tunable_groups: TunableGroups, mock_configs: List[dict]) -> Manua
service=None,
config={
"max_cycles": 2,
"cycle_tunable_values": mock_configs,
"tunable_values_cycle": mock_configs,
},
)

0 comments on commit 0a243cf

Please sign in to comment.