File size: 1,311 Bytes
246d201 |
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 |
import argparse
import os
from openhands.resolver.io_utils import load_single_resolver_output
def visualize_resolver_output(issue_number: int, output_dir: str, vis_method: str):
output_jsonl = os.path.join(output_dir, 'output.jsonl')
resolver_output = load_single_resolver_output(output_jsonl, issue_number)
if vis_method == 'json':
print(resolver_output.model_dump_json(indent=4))
else:
raise ValueError(f'Invalid visualization method: {vis_method}')
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Visualize a patch.')
parser.add_argument(
'--issue-number',
type=int,
required=True,
help='Issue number to send the pull request for.',
)
parser.add_argument(
'--output-dir',
type=str,
default='output',
help='Output directory to write the results.',
)
parser.add_argument(
'--vis-method',
type=str,
default='json',
choices=['json'],
help='Method to visualize the patch [json].',
)
my_args = parser.parse_args()
visualize_resolver_output(
issue_number=my_args.issue_number,
output_dir=my_args.output_dir,
vis_method=my_args.vis_method,
)
|