aeb516494daa207720b428072ab49eeea7bfff75,scripts/tests/test_scripts.py,,test_sampling,#Any#,115

Before Change

    if method == "beam_search":
        args.append("--use-beam-search")
    if method == "sampling":
        args.extend(["--use-sampling", "--temperature", "1.0"])
    process = subprocess.check_call([sys.executable, "./scripts/text_generation/sequence_sampling.py"]
                                     + args)
    time.sleep(5)

After Change

def test_sampling(method):
    args = ["--bos", "I love it", "--beam-size", "2", "--print-num", "1", "--gpu", "0"]
    if method == "beam_search":
        args.insert(0, "beam-search")
        args.extend(["--k", "50"])
    if method == "sampling":
        args.insert(0, "random-sample")
        args.extend(["--temperature", "1.0"])

Instances