onnxruntime/tools/ci_build/update_tsaoptions.py
2022-09-23 18:46:02 -07:00

19 lines
533 B
Python

# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
import json
import os
from pathlib import Path
SCRIPT_DIR = Path(os.path.dirname(os.path.realpath(__file__)))
REPO_DIR = SCRIPT_DIR.parent.parent
with (REPO_DIR / ".config" / "tsaoptions.json").open() as f:
data = json.load(f)
buildNumber = os.getenv("BUILD_BUILDNUMBER")
if buildNumber is not None:
data["buildNumber"] = buildNumber
with (REPO_DIR / ".config" / "tsaoptions.json").open(mode="w") as f:
json.dump(data, f)