def hesbo_branin()

in figs/fig_S1.py [0:0]

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

def hesbo_branin(X, mode):
    # In 2-D, HESBO has 3 possible embeddings:
    # Full rank, span x1 and x2
    # x1=x2
    # x1=-x2
    Y = X.copy()
    if mode == 1:
        pass  # All is well here!
    elif mode == 2:
        Y[:, 0] = Y[:, 1]
    elif mode == 3:
        Y[:, 0] = -Y[:, 1]
    return branin_centered(Y)