File size: 2,240 Bytes
adc02fa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 | #!/usr/bin/env python
from __future__ import annotations
import argparse
import os
import sys
from pathlib import Path
PROJECT_ROOT = Path(__file__).resolve().parents[1]
if str(PROJECT_ROOT) not in sys.path:
sys.path.insert(0, str(PROJECT_ROOT))
from dovla_cil.tasks.predicates import SUPPORTED_PREDICATES # noqa: E402
from dovla_cil.utils.io import write_jsonl
from dovla_cil.utils.seeding import seed_everything
from dovla_cil.vlm.client import VLMClient
from dovla_cil.vlm.task_generator import (
TaskGenerationRequest,
TaskGenerator,
default_task_generation_request,
)
def main(argv: list[str] | None = None) -> int:
parser = argparse.ArgumentParser(description="Generate VLM-assisted DoVLA-CIL task specs.")
parser.add_argument("--num-tasks", type=int, default=8)
parser.add_argument(
"--families",
nargs="*",
default=None,
help="Task families as space-separated or comma-separated values.",
)
parser.add_argument("--out", type=Path, default=Path("outputs/tasks.jsonl"))
parser.add_argument("--mock", action="store_true", help="Use deterministic mock generation.")
parser.add_argument("--seed", type=int, default=17)
args = parser.parse_args(argv)
seed_everything(args.seed)
if args.mock:
os.environ["OPENCLAUDE_MOCK"] = "1"
defaults = default_task_generation_request(num_tasks=args.num_tasks)
families = _parse_list(args.families) or defaults.families
request = TaskGenerationRequest(
num_tasks=args.num_tasks,
families=families,
object_categories=defaults.object_categories,
relations=sorted(SUPPORTED_PREDICATES),
constraints=[f"seed={args.seed}"],
)
generator = TaskGenerator(VLMClient())
tasks = generator.generate_tasks(request)
write_jsonl((task.to_dict() for task in tasks), args.out)
print(f"Wrote {len(tasks)} tasks to {args.out}")
return 0
def _parse_list(values: list[str] | None) -> list[str]:
if not values:
return []
parsed: list[str] = []
for value in values:
parsed.extend(item.strip() for item in value.split(",") if item.strip())
return parsed
if __name__ == "__main__":
raise SystemExit(main())
|