prioritize.py

import argparse, json
from explainable_test_prioritizer.modeling import prioritize_tests
from explainable_test_prioritizer.explainability import build_explanations
parser = argparse.ArgumentParser()
parser.add_argument("--input", required=True)
parser.add_argument("--model-dir", default="models")
parser.add_argument("--output-dir", default="outputs")
parser.add_argument("--top-k", type=int, default=None)
args = parser.parse_args()
ranked, summary = prioritize_tests(args.input, model_dir=args.model_dir, output_dir=args.output_dir, top_k=args.top_k)
exp = build_explanations(args.input, model_dir=args.model_dir, output_dir=args.output_dir, top_k=min(10, len(ranked)))
print(json.dumps({"summary": summary, "explanations": exp}, indent=2))