diff --git a/tests/test_bayesian_optimizer_optuna.py b/tests/test_bayesian_optimizer_optuna.py index b9e305e..e34d4e3 100644 --- a/tests/test_bayesian_optimizer_optuna.py +++ b/tests/test_bayesian_optimizer_optuna.py @@ -93,4 +93,3 @@ def test_real_optuna_runs_bounded_warm_start_and_restores_model(monkeypatch, pea assert model.forward_calls >= 5 # reference, generation and KL for each trial for name, tensor in model.state_dict().items(): torch.testing.assert_close(tensor, original[name], rtol=0, atol=0) -