import json
from pathlib import Path
CP = Path("/Users/manta/Documents/Projects/TheRoad-I1/projects/b0ad5c18-5140-4022-a2b0-805de0e5a385/checkpoints/episodes/0d30302a-51c8-4ac9-98e2-6cd9180512bf")
groups = json.loads((CP / "background_render" / "manifest.json").read_text())["data"]["groups"]
by_loc = {}
for bg_id, gr in groups.items():
    loc = gr.get("location_id")
    by_loc.setdefault(loc, []).append((bg_id, gr.get("shot_ids") or gr.get("shots") or []))
for loc in sorted(by_loc):
    if loc != "L04":
        continue
    for bg_id, sids in by_loc[loc]:
        print(loc, bg_id, sids)