mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Summary: This involves another purely cosmetic (ordering) change to the `config.yml` to facilitate simpler logic. Other changes: * add some review feedback as comments * exit with nonzero status on config.yml mismatch * produce a diagram for pytorch builds Pull Request resolved: https://github.com/pytorch/pytorch/pull/17427 Differential Revision: D14197618 Pulled By: kostmo fbshipit-source-id: 267439d3aa4c0a80801adcde2fa714268865900e
40 lines
1.0 KiB
Python
Executable File
40 lines
1.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
|
|
import generate_config_yml
|
|
|
|
|
|
CHECKED_IN_FILE = "config.yml"
|
|
REGENERATION_SCRIPT = "regenerate.sh"
|
|
|
|
PARENT_DIR = os.path.basename(os.path.dirname(os.path.abspath(__file__)))
|
|
README_PATH = os.path.join(PARENT_DIR, "README.md")
|
|
|
|
ERROR_MESSAGE_TEMPLATE = """
|
|
The checked-in CircleCI "%s" file does not match what was generated by the scripts.
|
|
Please re-run the "%s" script in the "%s" directory and commit the result. See "%s" for more information.
|
|
"""
|
|
|
|
|
|
def check_consistency():
|
|
|
|
_, temp_filename = tempfile.mkstemp("-generated-config.yml")
|
|
|
|
with open(temp_filename, "w") as fh:
|
|
generate_config_yml.stitch_sources(fh)
|
|
|
|
try:
|
|
subprocess.check_call(["cmp", temp_filename, CHECKED_IN_FILE])
|
|
except subprocess.CalledProcessError:
|
|
sys.exit(ERROR_MESSAGE_TEMPLATE % (CHECKED_IN_FILE, REGENERATION_SCRIPT, PARENT_DIR, README_PATH))
|
|
finally:
|
|
os.remove(temp_filename)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
check_consistency()
|