fixed issue in which train_test_split was not properly imported due to sklearn being imported in parts