Hi @Avni2
Please keep in mind that numpy does not allow direct comparison of arrays using == without handling the possibility of multiple elements. So, please check your value types that are being compared or you can use ‘np.allclose’ or ‘np.array_equal’ for comparing your arrays.
Hope it helps! Feel free to ask if you need further assistance.
The evidence suggests your code is not correct.