Fix to the test_cross_validation.py script

Modified the lstsq Op
The 3rd output is now an numpy array instead of an int.
We now check if the 1st output contains elements of type float instead of type int.