Skip to content

Commit 32e45c4

Browse files
remove excess type checking
1 parent c5310c7 commit 32e45c4

2 files changed

Lines changed: 9 additions & 9 deletions

File tree

src/blop/ax/optimizer.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -230,22 +230,22 @@ def _apply_parameter_update(
230230
231231
Raises
232232
------
233-
ValueError
233+
TypeError
234234
If the provided value does not match the expected type for the parameter.
235235
"""
236236
parameter = self._client._experiment.parameters[parameter_name]
237237
if isinstance(parameter, RangeParameter):
238238
if not isinstance(value, tuple):
239-
raise ValueError(f"{RangeParameter.__name__} only accepts tuples of length 2, but got: {value}")
239+
raise TypeError(f"{RangeParameter.__name__} only accepts tuples of length 2, but got: {value}")
240240
original_range_values[parameter_name] = (parameter.lower, parameter.upper)
241241
parameter.update_range(*value)
242242
elif isinstance(parameter, ChoiceParameter):
243-
if not isinstance(value, list) or not all(isinstance(v, type(parameter.values[0])) for v in value):
244-
raise ValueError(f"{ChoiceParameter.__name__} only accepts list of items, but got: {value}")
243+
if not isinstance(value, list):
244+
raise TypeError(f"{ChoiceParameter.__name__} only accepts list of items, but got: {value}")
245245
original_choice_values[parameter_name] = parameter.values
246246
parameter.set_values(value)
247247
else:
248-
raise ValueError(f"Expected RangeParameter or ChoiceParameter, but got {parameter}")
248+
raise TypeError(f"Expected RangeParameter or ChoiceParameter, but got {parameter}")
249249

250250
def _rollback_parameter_updates(
251251
self,

src/blop/tests/ax/test_optimizer.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -199,13 +199,13 @@ def test_ax_optimizer_reconfigurable_search_space():
199199
with pytest.raises(KeyError):
200200
optimizer._reconfigure_search_space({"x4": (-4, 4)})
201201
# ChoiceParameter expects a list
202-
with pytest.raises(ValueError):
202+
with pytest.raises(TypeError):
203203
optimizer._reconfigure_search_space({"x3": 3})
204-
# ChoiceParameter expects a list of single type
204+
# ChoiceParameter expects a list of types castable to parameter_type
205205
with pytest.raises(ValueError):
206-
optimizer._reconfigure_search_space({"x3": ["2", 5, 3.6]})
206+
optimizer._reconfigure_search_space({"x3": ["2", "Hello", 3.6]})
207207
# RangeParameter expects a tuple
208-
with pytest.raises(ValueError):
208+
with pytest.raises(TypeError):
209209
optimizer._reconfigure_search_space({"x1": 3})
210210

211211
# Changing the serach space should reflect in parameter state

0 commit comments

Comments
 (0)