| 1 | # SPDX-License-Identifier: Apache-2.0 |
| 2 | # SPDX-FileCopyrightText: Copyright contributors to the vLLM project |
| 3 | |
| 4 | """Regenerate tests/watermarking/watermarking_goldens.json.""" |
| 5 | |
| 6 | import json |
| 7 | from pathlib import Path |
| 8 | |
| 9 | from tests.watermarking.golden_candidates import golden_payload, validate_golden_guards |
| 10 | |
| 11 | GOLDENS_PATH = Path(__file__).with_name("watermarking_goldens.json") |
| 12 | |
| 13 | |
| 14 | def main() -> None: |
| 15 | payload = golden_payload() |
| 16 | validate_golden_guards(payload["candidates"]) |
| 17 | GOLDENS_PATH.write_text( |
| 18 | json.dumps(payload, indent=2, sort_keys=True) + "\n", encoding="utf-8" |
| 19 | ) |
| 20 | print(f"wrote {len(payload['candidates'])} candidates to {GOLDENS_PATH}") |
| 21 | |
| 22 | |
| 23 | if __name__ == "__main__": |
| 24 | main() |