Skip to content

Commit

Permalink
test: Fixed test to allow reproduced results to be better than the FL…
Browse files Browse the repository at this point in the history
…AML results, when LGBM earlystopping is involved
  • Loading branch information
Daniel Grindrod authored and Daniel Grindrod committed Oct 31, 2024
1 parent a209d7a commit d8dfdc7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/automl/test_regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -382,7 +382,7 @@ def test_reproducibility_of_lgbm_regression_model():
fit_kwargs=None,
free_mem_ratio=0,
)
assert pytest.approx(val_loss_flaml) == reproduced_val_loss
assert pytest.approx(val_loss_flaml) == reproduced_val_loss or val_loss_flaml > reproduced_val_loss


@pytest.mark.parametrize(
Expand Down

0 comments on commit d8dfdc7

Please sign in to comment.