6880ee0c2c46573cfc77444e2970aa8695140076,thumt/scripts/shuffle_corpus.py,,main,#,24

Before Change

    for fdr, fdw in zip(stream, newstream):
        fdr.close()
        fdw.close()


if __name__ == "__main__":
    parsed_args = parseargs()

After Change

    if args.num_shards == 1:
        newstream = [[open(item + suffix, "w") for item in name]]
    else:
        newstream = [[open(item + "-%s-of-%s" % (i, args.num_shards), "w")
                      for item in name] for i in range(args.num_shards)]

    for idx in indices.tolist():
        lines = [item[idx] for item in data]

Instances