def encode_csv()

in src/sagemaker_huggingface_inference_toolkit/decoder_encoder.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

def encode_csv(content, accept_type=None):
    """Convert the result of a transformers pipeline to CSV.
    Args:
        content (dict | list): result of transformers pipeline.
    Returns:
        (str): object serialized to CSV
    """
    stream = StringIO()
    if not isinstance(content, list):
        content = list(content)

    column_header = content[0].keys()
    writer = csv.DictWriter(stream, column_header)

    writer.writeheader()
    writer.writerows(content)
    return stream.getvalue()