@@ -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)
429428def 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