TODO: fix so that the regex can handle trailing commas, in black style add a space before the :returns: line