def METHOD_NAME():
    parser = make_parser()
    args = options.parse_args_and_arch(parser)
    dataset = load_dataset_task(args)
    random_dataset = randomly_sample_subset(dataset)
    short_dataset = get_short_data_subset(dataset)
    long_dataset = get_long_data_subset(dataset)
    if args.dataset_save_token:
        args.dataset_save_token = f"_{args.dataset_save_token}_"
    if args.dataset_save_dir:
        save_dataset_npy(
            random_dataset,
            f"{args.dataset_save_dir}/random_dataset{args.dataset_save_token}w_ids.npy",
        )
        save_dataset_npy(
            short_dataset,
            f"{args.dataset_save_dir}/short_dataset{args.dataset_save_token}w_ids.npy",
        )
        save_dataset_npy(
            long_dataset,
            f"{args.dataset_save_dir}/long_dataset{args.dataset_save_token}w_ids.npy",
        )