def create_prompt_for_length()

in scripts/benchmark_pipelines.py [0:0]

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

def create_prompt_for_length(batch: int, length: int) -> Union[str, List[str]]:
    tokens = ["I"] * length
    tokens = " ".join(tokens)
    if batch == 1:
        return tokens
    return [tokens] * batch