Fix small typo that makes code not run 
Fix small typo that makes code not run 

Occurs when trying to split dataset into test and train with 50% ratio
tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets