update_doc.py 999 Bytes
Newer Older
Merve Noyan's avatar
Merve Noyan committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import subprocess
import argparse


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--check", action="store_true")

    args = parser.parse_args()

    output = subprocess.check_output(["text-generation-launcher", "--help"]).decode("utf-8")
    final_doc = f"# Text-generation-launcher arguments\n```\n{output}\n```"

    filename = "docs/source/basic_tutorials/launcher.md"
    if args.check:
        with open(filename, "r") as f:
            doc = f.read()
            if doc != final_doc:

                tmp = "launcher.md"
                with open(tmp, "w") as g:
                    g.write(final_doc)
                diff = subprocess.run(["diff",tmp, filename], capture_output=True).stdout.decode("utf-8")
                print(diff)
                raise Exception("Doc is not up-to-date, run `python update_doc.py` in order to update it")
    else:
        with open(filename, "w") as f:
            f.write(final_doc)

if __name__ == "__main__":
    main()