def with_incremental_state()

in esm/multihead_attention.py [0:0]

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

def with_incremental_state(cls):
    cls.__bases__ = (FairseqIncrementalState,) + tuple(
        b for b in cls.__bases__ if b != FairseqIncrementalState
    )
    return cls