| """Build the matched-random control by permuting residual masses per trajectory.""" |
|
|
| from __future__ import annotations |
|
|
| import argparse |
| import json |
| from pathlib import Path |
|
|
| from starVLA.dataloader.infoflow_sampler import ValidAnchorManifest |
| from starVLA.dataloader.temporal_proposal import TemporalProposal |
| from starVLA.scoring.matched_random import build_matched_random_proposal |
|
|
|
|
| def main() -> None: |
| parser = argparse.ArgumentParser() |
| parser.add_argument("--manifest", type=Path, required=True) |
| parser.add_argument("--residual-proposal", type=Path, required=True) |
| parser.add_argument("--output", type=Path, required=True) |
| parser.add_argument("--permutation-seed", type=int, required=True) |
| args = parser.parse_args() |
|
|
| manifest = ValidAnchorManifest.load(args.manifest) |
| residual = TemporalProposal.load(args.residual_proposal, manifest) |
| matched = build_matched_random_proposal( |
| manifest=manifest, |
| residual_proposal=residual, |
| permutation_seed=args.permutation_seed, |
| ) |
| matched.save(args.output) |
| print( |
| json.dumps( |
| { |
| "status": "complete", |
| "manifest_sha256": manifest.sha256, |
| "source_proposal_sha256": residual.sha256, |
| "matched_random_proposal_sha256": matched.sha256, |
| "permutation_seed": args.permutation_seed, |
| "output": str(args.output.resolve()), |
| }, |
| indent=2, |
| sort_keys=True, |
| ) |
| ) |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
