diff --git a/.github/dependabot.yml b/.github/dependabot.yml index a7c46a2349b..145d52de746 100644 --- a/.github/dependabot.yml +++ b/.github/dependabot.yml @@ -11,3 +11,8 @@ updates: versions: - ">= 0.0.a" - "< 0.1" +- package-ecosystem: "github-actions" + directory: "/" + schedule: + interval: daily + open-pull-requests-limit: 10 diff --git a/tests/training/learning_rate_schedulers/cosine_test.py b/tests/training/learning_rate_schedulers/cosine_test.py index 3cff5f62b99..1013fa4a91d 100644 --- a/tests/training/learning_rate_schedulers/cosine_test.py +++ b/tests/training/learning_rate_schedulers/cosine_test.py @@ -130,7 +130,7 @@ def test_schedules(self): lrs.append(optimizer.param_groups[0]["lr"]) for it, lr in lr_checks: - assert lrs[it] == lr, f"Iteration {it}: {lrs[it]} != {lr}" + assert lrs[it] == pytest.approx(lr), f"Iteration {it}: {lrs[it]} != {lr}" def test_schedules_with_save_and_resume(self): """Make sure scheduler will resume with the right state.""" @@ -171,4 +171,4 @@ def init_and_restore_scheduler( state = scheduler.state_dict() for it, lr in lr_checks: - assert lrs[it] == lr, f"Iteration {it}: {lrs[it]} != {lr}" + assert lrs[it] == pytest.approx(lr), f"Iteration {it}: {lrs[it]} != {lr}"