[brief] Forgot to test if the custom schedulers were registered.

[detailed] - Accidentally deleted them when I updated the tests.
marovira · May 28, 2024 · b795831 · b795831
1 parent ee0ed5d
commit b795831
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/test_scheduler.py b/test/test_scheduler.py
@@ -6,6 +6,7 @@ def test_registry(self, check_registry) -> None:
         check_registry(
             scheduler.SCHEDULER_REGISTRY,
             [
+                # Pytorch schedulers.
                 "LambdaLR",
                 "MultiplicativeLR",
                 "StepLR",
@@ -20,6 +21,9 @@ def test_registry(self, check_registry) -> None:
                 "CyclicLR",
                 "OneCycleLR",
                 "CosineAnnealingWarmRestarts",
+                # Custom schedulers.
+                "MultiStepRestartLR",
+                "CosineAnnealingRestartLR",
             ],
         )