def METHOD_NAME(cls) -> dict:
    default_ag_args = super().METHOD_NAME()
    extra_ag_args = {
        "problem_types": [BINARY, MULTICLASS],
    }
    default_ag_args.update(extra_ag_args)
    return default_ag_args