def get_norm()

in lib/util.py [0:0]

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

def get_norm(name, d, dtype=th.float32):
    if name == "none":
        return lambda x: x
    elif name == "layer":
        return tu.LayerNorm(d, dtype=dtype)
    else:
        raise NotImplementedError(name)