bfeab59160b5676b0a1a613181dc759a5be4a2e2,official/core/input_reader.py,InputReader,read,#InputReader#,220

Before Change

    elif self._num_files > 1:
      dataset = self._read_sharded_files(input_context)
    else:
      assert self._num_files == 1
      dataset = self._read_single_file(input_context)

    if self._cache:
      dataset = dataset.cache()

After Change

      dataset = self._read_tfds(input_context)
    elif len(self._matched_files) > 1:
      dataset = self._read_sharded_files(input_context)
    elif len(self._matched_files) == 1:
      dataset = self._read_single_file(input_context)
    else:
      raise ValueError("It is unexpected that `tfds_builder` is None and "
                       "there is also no `matched_files`.")

Instances