Skip to content

Commit 9102ce5

Browse files
authored
Mypy version fixups (#594)
1 parent 9a4f90f commit 9102ce5

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

mlos_core/mlos_core/optimizers/flaml_optimizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,8 +164,8 @@ def _get_next_config(self) -> dict:
164164
points_to_evaluate: list = []
165165
evaluated_rewards: list = []
166166
if len(self.evaluated_samples) > 0:
167-
evaluated_samples_list: list = [(s.config, s.score) for s in self.evaluated_samples.values()]
168-
points_to_evaluate, evaluated_rewards = list(zip(*evaluated_samples_list))
167+
points_to_evaluate = [s.config for s in self.evaluated_samples.values()]
168+
evaluated_rewards = [s.score for s in self.evaluated_samples.values()]
169169

170170
# Warm start FLAML optimizer
171171
self._suggested_config = None

mlos_core/mlos_core/spaces/adapters/llamatune.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -296,7 +296,7 @@ def _validate_special_param_values(self, special_param_values_dict: dict) -> Non
296296
tuple_list = [(value, self.DEFAULT_SPECIAL_PARAM_VALUE_BIASING_PERCENTAGE)]
297297
elif isinstance(value, tuple) and [type(v) for v in value] == [int, float]:
298298
# User specifies both special value and biasing percentage
299-
tuple_list = [value] # type: ignore[list-item]
299+
tuple_list = [value]
300300
elif isinstance(value, list) and value:
301301
if all(isinstance(t, int) for t in value):
302302
# User specifies list of special values

0 commit comments

Comments
 (0)