Dataset load from sqlite / split error

[mariosasko]

ds["col1"] returns a Python list, hence the error. To avoid it, use train_testvalid = ds.select_columns("col1").train_test_split(test_size=0.2) instead.