def METHOD_NAME(
    self,
    modifier_lambda: Callable[[], ScheduledUpdateModifier],
    model_lambda: Callable[[], keras.models.Model],
    steps_per_epoch: int,
):
    super().METHOD_NAME(modifier_lambda, framework=KERAS_FRAMEWORK)