| #!/usr/bin/env python3 | |
| # Copyright 2026 The IREE Authors | |
| # | |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. | |
| # See https://llvm.org/LICENSE.txt for license information. | |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | |
| """Refresh author-owned asset locks and catalog metadata with the installed SDK.""" | |
| import argparse | |
| import subprocess | |
| import sys | |
| from pathlib import Path | |
| def main(): | |
| parser = argparse.ArgumentParser(description=__doc__) | |
| parser.add_argument( | |
| "--cache", type=Path, help="reuse downloaded assets in this Loom cache" | |
| ) | |
| args = parser.parse_args() | |
| root = Path(__file__).absolute().parent | |
| cards = sorted(root.glob("model/*/model.json")) | |
| if not cards: | |
| parser.error("no model/*/model.json files found beside this script") | |
| for card in cards: | |
| command = [ | |
| sys.executable, | |
| "-m", | |
| "loom.model.fetch_main", | |
| str(card), | |
| "--update-lock", | |
| "--all-variants", | |
| "--no-input", | |
| ] | |
| if args.cache is not None: | |
| command.append("--cache=" + str(args.cache.absolute())) | |
| subprocess.run(command, check=True) | |
| subprocess.run( | |
| [ | |
| sys.executable, | |
| "-m", | |
| "loom.model.catalog", | |
| "--output=" + str(root / "catalog.json"), | |
| *[ | |
| "--card=" + card.relative_to(root).as_posix() + "=" + str(card) | |
| for card in cards | |
| ], | |
| ], | |
| check=True, | |
| ) | |
| if __name__ == "__main__": | |
| main() | |