examples/legacy/multiple_choice/utils_multiple_choice.py/TFMultipleChoiceDataset/get_dataset
class TFMultipleChoiceDataset:        def get_dataset(self):
            self.dataset = self.dataset.apply(tf.data.experimental.assert_cardinality(len(self.features)))

            return self.dataset