(package: dict[str, Any], checkpoint_path: Path, signature: str)
| 1489 | handle.write(chunk) |
| 1490 | written += len(chunk) |
| 1491 | if expected_size is not None and written != expected_size: |
| 1492 | raise RuntimeError(f"downloaded size mismatch for {target}: {written} != {expected_size}") |
| 1493 | return written |
| 1494 | |
| 1495 | |
| 1496 | def validate_required_files(package: ModelPackage, root: Path) -> None: |
| 1497 | missing = [relative for relative in package.required_files if not (root / relative).exists()] |
| 1498 | if missing: |
| 1499 | raise RuntimeError(f"installed package is missing required files: {missing}") |
| 1500 | |
| 1501 | |
| 1502 | def validate_composite_required_files(package: ModelPackage, staged_root: Path, final_root: Path) -> None: |
| 1503 | missing = [ |
| 1504 | relative |
| 1505 | for relative in package.required_files |
| 1506 | if not normalized_join(staged_root, relative).exists() and not normalized_join(final_root, relative).exists() |
| 1507 | ] |
| 1508 | if missing: |
no test coverage detected