Leetcode Single Number Python Compiler With Sklearn Train Test Split