I got "ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()"