onnxruntime

1
# generate_skip_doc_change.py
2
import os
3
from pathlib import Path
4

5
from jinja2 import Environment, FileSystemLoader, Template
6

7
GITHUB_DIR = Path(__file__).resolve().parent.parent
8

9

10
class Skipped_Workflow:  # noqa: N801
11
    def __init__(self, workflow_name: str, job_names: list, output_file_name: str):
12
        self.workflow_name = workflow_name
13
        self.job_names = job_names
14
        self.fake_file_name = output_file_name
15

16

17
WIN_GPU_CI_WORKFLOW = Skipped_Workflow(
18
    workflow_name="Windows GPU CI Pipeline",
19
    job_names=[
20
        "cuda build_x64_RelWithDebInfo",
21
        "dml build_x64_RelWithDebInfo",
22
        "training build_x64_RelWithDebInfo",
23
        "kernelDocumentation build_x64_RelWithDebInfo",
24
    ],
25
    output_file_name=str(GITHUB_DIR.joinpath("workflows/generated_fake_win_gpu_ci.yml")),
26
)
27

28

29
def generate_fake_ci_yaml(template: Template, workflow: Skipped_Workflow):
30
    py_file_name = os.path.basename(__file__)
31
    content = template.render(
32
        python_file_name=py_file_name, ci_workflow_name=workflow.workflow_name, job_names=workflow.job_names
33
    )
34

35
    filename = workflow.fake_file_name
36
    with open(filename, mode="w", encoding="utf-8") as output_file:
37
        output_file.write(content)
38
        if content[-1] != "\n":
39
            output_file.write("\n")
40
        print(f"... wrote {filename}")
41

42

43
def main() -> None:
44
    environment = Environment(loader=FileSystemLoader(str(GITHUB_DIR.joinpath("workflows/"))))
45
    template = environment.get_template("skip-doc-change.yml.j2")
46
    skipped_workflows = [WIN_GPU_CI_WORKFLOW]
47
    [generate_fake_ci_yaml(template, workflow) for workflow in skipped_workflows]
48

49

50
if __name__ == "__main__":
51
    main()
52