77from kernel_tuner .searchspace import Searchspace
88from kernel_tuner .strategies import common
99from kernel_tuner .strategies .common import CostFunc
10- from kernel_tuner .util import StopCriterionReached
10+ from kernel_tuner .util import StopCriterionReached , TuningBudget
1111
1212try :
1313 from mock import Mock
@@ -30,7 +30,7 @@ def fake_runner():
3030
3131def test_cost_func ():
3232 x = [1 , 4 ]
33- tuning_options = Options (scaling = False , snap = False , tune_params = tune_params ,
33+ tuning_options = Options (tune_params = tune_params , budget = TuningBudget () ,
3434 restrictions = None , strategy_options = {}, cache = {}, unique_results = {},
3535 objective = "time" , objective_higher_is_better = False , metrics = None )
3636 runner = fake_runner ()
@@ -41,7 +41,7 @@ def test_cost_func():
4141 # check if restrictions are properly handled
4242 def restrictions (x , y ):
4343 return False
44- tuning_options = Options (scaling = False , snap = False , tune_params = tune_params ,
44+ tuning_options = Options (tune_params = tune_params , budget = TuningBudget () ,
4545 restrictions = restrictions , strategy_options = {},
4646 verbose = True , cache = {}, unique_results = {},
4747 objective = "time" , objective_higher_is_better = False , metrics = None )
0 commit comments