def METHOD_NAME(
    self,
    METHOD_NAME: dict,
    model: torch.nn.Module,
    encoder,
    valid_df: pd.DataFrame = None,
    test_df: pd.DataFrame = None,
):
    return ScenePredictionTask(model, encoder, **METHOD_NAME)