Further simplify the typing in the example

Signed-off-by: Fabrice Normandin <[email protected]>

project/algorithms/example.py

@@ -6,7 +6,7 @@
 import dataclasses
 import functools
 from logging import getLogger
-from typing import Annotated, Any, Literal
+from typing import Any, Literal
 
 import pydantic
 import torch
@@ -26,8 +26,6 @@
 
 logger = getLogger(__name__)
 
-LRSchedulerConfig = Annotated[Any, pydantic.Field(default_factory=CosineAnnealingLRConfig)]
-
 
 class ExampleAlgorithm(LightningModule):
     """Example learning algorithm for image classification."""
@@ -37,11 +35,7 @@ class HParams:
         """Hyper-Parameters."""
 
         # Arguments to be passed to the LR scheduler.
-        lr_scheduler: LRSchedulerConfig = dataclasses.field(
-            default=CosineAnnealingLRConfig(T_max=85, eta_min=1e-5),
-            metadata={"omegaconf_ignore": True},
-        )
-
+        lr_scheduler: Any = CosineAnnealingLRConfig(T_max=85, eta_min=1e-5)
         lr_scheduler_interval: Literal["step", "epoch"] = "epoch"
 
         # Frequency of the LR scheduler. Set to 0 to disable the lr scheduler.