def main()

in utils/ryzenai/generate_operators_baseline.py [0:0]


def main():
    parser = argparse.ArgumentParser(description="Process JSON files in folders.")
    parser.add_argument("input_folder", help="Path to the input folder")
    parser.add_argument("output_json", help="Path to the output JSON file")
    args = parser.parse_args()

    input_folder = args.input_folder
    output_json = args.output_json

    if not os.path.exists(input_folder):
        print(f"Error: Input folder '{input_folder}' does not exist.")
        return

    result_dict = process_folder(input_folder)
    sorted_result = dict(sorted(result_dict.items()))

    with open(output_json, "w") as output_file:
        json.dump(sorted_result, output_file, indent=2)

    print(f"Processed successfully. Result saved to '{output_json}'.")