def METHOD_NAME(
    self,
    modifier_lambda: Callable[[], ScheduledModifier],
    model_lambda: Callable[[], Module],
    optim_lambda: Callable[[Module], Optimizer],
    test_epoch: float,  # noqa: F811
    test_steps_per_epoch: int,  # noqa: F811
):
    super().METHOD_NAME(
        modifier_lambda,
        model_lambda,
        optim_lambda,
        test_epoch,
        test_steps_per_epoch,
        distillation_teacher=model_lambda(),
    )