Skip to content

Commit aec5a95

Browse files
committed
Adding params for EvoTorch
1 parent 99dad48 commit aec5a95

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/functional/test_evolutionary.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ def test_evotorch_cmaes(
219219
method="cmaes",
220220
directions=["minimize"],
221221
calibration_func_kwargs=dict(t=observed_data.day),
222-
method_kwargs=dict(),
222+
method_kwargs=dict(stdev_init=1),
223223
)
224224

225225
calibrator = get_calibrator(
@@ -253,7 +253,7 @@ def test_evotorch_cosyne(
253253
method="cosyne",
254254
directions=["minimize"],
255255
calibration_func_kwargs=dict(t=observed_data.day),
256-
method_kwargs=dict(),
256+
method_kwargs=dict(popsize=100, tournament_size=20, mutation_stdev=1),
257257
)
258258

259259
calibrator = get_calibrator(

0 commit comments

Comments
 (0)