def METHOD_NAME(ds_args, tokenizer):
    if ds_args.block_lm:
        ds_args.data_set_type = "Block"
    if ds_args.sentinel_token:
        ds_args.add_sentinel_token = ds_args.max_position_embeddings
    return ds_args