Fix TFDS num_example

PiperOrigin-RevId: 388874067

uncertainty_baselines/datasets/base.py

@@ -38,18 +38,6 @@
     types.Features]
 
 
-def _absolute_split_len(absolute_split, dataset_splits):
-  if absolute_split.from_ is None:
-    start = 0
-  else:
-    start = absolute_split.from_
-  if absolute_split.to is None:
-    end = dataset_splits[absolute_split.splitname].num_examples
-  else:
-    end = absolute_split.to
-  return end - start
-
-
 def get_validation_percent_split(
     dataset_builder,
     validation_percent,
@@ -210,13 +198,6 @@ def split(self):
 
   @property
   def num_examples(self):
-    if isinstance(self._split, tfds.core.ReadInstruction):
-      absolute_split = self._split.to_absolute(
-          {
-              name: self.tfds_info.splits[name].num_examples
-              for name in self.tfds_info.splits.keys()
-          })[0]
-      return _absolute_split_len(absolute_split, self.tfds_info.splits)
     return self.tfds_info.splits[self._split].num_examples
 
   def _create_process_example_fn(self) -> Optional[PreProcessFn]: