I am having trouble to find the line of code producing the FutureWarning message: elementwise comparison failed.
There are other questions in SO that describe the pyt