def adjust_length_to_model()

in model/mm_dst/gpt2_dst/scripts/run_retrieval.py [0:0]

• 8 lines of code
• 5 McCabe index (conditional complexity)

def adjust_length_to_model(length, max_sequence_length):
    if length < 0 and max_sequence_length > 0:
        length = max_sequence_length
    elif 0 < max_sequence_length < length:
        length = max_sequence_length  # No generation bigger than model size
    elif length < 0:
        length = MAX_LENGTH  # avoid infinite loop
    return length