Skip to content

Commit ae06489

Browse files
committed
Update scenario to be able to support parameters for multiple bindings to same label.
1 parent 50bbe0f commit ae06489

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

src/hg_systematic/strategy/scenario.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,9 @@ def set_parameters(label: str | Callable, **kwargs):
5757
if not isinstance(label, str):
5858
label = label.signature.name
5959
overloads = _LBL_TO_OVERLOAD[label]
60-
if len(overloads) != 1:
61-
raise ValueError(f"Expected 1 overload for scenario {label}, got {len(overloads)}.")
60+
assert len(overloads) >= 1, f"Expected at least 1 overload for scenario label {label}"
61+
# All overloads using the same scenario label must use the same parameters,
62+
# so just pick one and run with it
6263
overload = next(iter(overloads))
6364
parameters = _SCENARIOS[overload][label]
6465
if len(parameters) != len(kwargs) or set(kwargs.keys()) - set(parameters):

0 commit comments

Comments
 (0)