#!/usr/bin/env python
"""Download the VBench weights the eight protocol dimensions need.

Everything lands in $VBENCH_CACHE_DIR so the run does not depend on whatever a
user's ~/.cache happens to hold.  `local=True` is used wherever VBench supports
it, so CLIP and DINO come from the cache directory too rather than from the clip
package's own download path.
"""
import os
import sys

DIMS = ["subject_consistency", "background_consistency", "motion_smoothness",
        "dynamic_degree", "aesthetic_quality", "imaging_quality", "scene",
        "overall_consistency"]

def main():
    from vbench.utils import init_submodules, CACHE_DIR
    print("CACHE_DIR =", CACHE_DIR, flush=True)
    for d in DIMS:
        print(f"--- {d} ---", flush=True)
        try:
            got = init_submodules([d], local=True, read_frame=False)
            print(f"    ok: {got.get(d)}", flush=True)
        except Exception as e:
            print(f"    FAILED: {type(e).__name__}: {e}", flush=True)
    print("done", flush=True)

if __name__ == "__main__":
    sys.exit(main())