Skip to content

Commit 3d7809e

Browse files
committed
Remove unnecessary extra_kwargs
1 parent 3e8211d commit 3d7809e

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

tests/test_acquisition.py

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -407,12 +407,12 @@ def verify_optimizers_match(optimizer1, optimizer2):
407407

408408

409409
@pytest.mark.parametrize(
410-
("acquisition_fn_factory", "state_filename", "extra_kwargs"),
410+
("acquisition_fn_factory", "state_filename"),
411411
[
412-
(lambda: UpperConfidenceBound(kappa=2.576), "ucb_state.json", {}),
413-
(lambda: ProbabilityOfImprovement(xi=0.01), "pi_state.json", {}),
414-
(lambda: ExpectedImprovement(xi=0.01), "ei_state.json", {}),
415-
(lambda: ConstantLiar(base_acquisition=UpperConfidenceBound(kappa=2.576)), "cl_state.json", {}),
412+
(lambda: UpperConfidenceBound(kappa=2.576), "ucb_state.json"),
413+
(lambda: ProbabilityOfImprovement(xi=0.01), "pi_state.json"),
414+
(lambda: ExpectedImprovement(xi=0.01), "ei_state.json"),
415+
(lambda: ConstantLiar(base_acquisition=UpperConfidenceBound(kappa=2.576)), "cl_state.json"),
416416
(
417417
lambda: GPHedge(
418418
base_acquisitions=[
@@ -422,12 +422,11 @@ def verify_optimizers_match(optimizer1, optimizer2):
422422
]
423423
),
424424
"gphedge_state.json",
425-
{},
426425
),
427426
],
428427
)
429428
def test_integration_acquisition_functions(
430-
acquisition_fn_factory, state_filename, extra_kwargs, target_func_x_and_y, pbounds, tmp_path
429+
acquisition_fn_factory, state_filename, target_func_x_and_y, pbounds, tmp_path
431430
):
432431
"""Parametrized integration test for acquisition functions."""
433432
acquisition_function = acquisition_fn_factory()
@@ -438,7 +437,6 @@ def test_integration_acquisition_functions(
438437
acquisition_function=acquisition_function,
439438
random_state=1,
440439
verbose=0,
441-
**extra_kwargs,
442440
)
443441
optimizer.maximize(init_points=2, n_iter=3)
444442

@@ -451,7 +449,6 @@ def test_integration_acquisition_functions(
451449
acquisition_function=acquisition_fn_factory(),
452450
random_state=1,
453451
verbose=0,
454-
**extra_kwargs,
455452
)
456453
new_optimizer.load_state(state_path)
457454

0 commit comments

Comments
 (0)