U <cp@sddlZddlZddlZddlmZmZmZmZmZddl m Z ee j e j e j e jfZeedddZeedddZeeeefefeegefeeeefefd d d Zeed d dZdS)N)AnyCallableDictListUnion)sarif)sreturncCstdd|S)Nz([A-Z])z_\1)resublowerrrT/tmp/pip-unpacked-wheel-gikjz4vx/torch/onnx/_internal/diagnostics/infra/formatter.py_camel_case_to_snake_casesrcCs |ddS)N-_)replacer rrrkebab_case_to_snake_casesr)objectconvertr cspt|ts|Si}|D]P\}}|}t|tr@t|}n"t|tr^fdd|D}n|}|||<q|S)agConvert and update keys in a dictionary with "convert". Any value that is a dictionary will be recursively updated. Any value that is a list will be recursively searched. Args: object: The object to update. convert: The function to convert the keys, e.g. `kebab_case_to_snake_case`. Returns: The updated object. csg|]}t|qSr) _convert_key).0elemrrr 1sz _convert_key..) isinstanceritemsrr)rrZnew_dictkvZnew_kZnew_vrrrrs     r) attr_cls_objr cCs"t|}t|t}tj|ddS)N)indent) dataclassesZasdictrrjsondumps)r dictrrr sarif_to_json8s  r')r#r$r typingrrrrrZ&torch.onnx._internal.diagnostics.infrarZSarifLogZRunZReportingDescriptorZResultZ _SarifClassstrrrrr'rrrrs$