tensorflow dataset can't figure out number of train steps in epoch after filtering

Question

I create a tensorflow dataset then filter it to split it to train and test sets as below:

test_index, train_index = split_data_for_train_and_test