utils

utils

Functions:

Name	Description
input_to_json_records	Try and convert python objects to a list of Fields

input_to_json_records(in_data)

Try and convert python objects to a list of Fields

Source code in src/nemo_safe_synthesizer/pii_replacer/ner/utils.py

def input_to_json_records(in_data: InData) -> list[JSONRecord]:
    """Try and convert python objects to a list of Fields"""
    if isinstance(in_data, JSONRecord):
        return [in_data]
    if isinstance(in_data, (str, dict)):
        return [JSONRecord(in_data)]
    if isinstance(in_data, list):
        out = []
        for record in in_data:
            if isinstance(record, JSONRecord):
                out.append(record)
            else:
                out.append(JSONRecord(record))
        return out
    raise TypeError("Input data not supported.")