In line 48: expected_improvement[sigma == 0.0] == 0.0 It should be: expected_improvement[sigma == 0.0] = 0.0