7d5a866a254cb9517ed24818b92e6f617fa4ed24,nucleus/io/genomics_reader.py,TFRecordReader,__init__,#TFRecordReader#Any#Any#Any#,131

Before Change

    self.header = None

    if not tf_options:
      compressed = input_path.endswith(".gz")
      tf_options = python_io.TFRecordOptions(
          python_io.TFRecordCompressionType.GZIP if compressed else
          python_io.TFRecordCompressionType.NONE)
    self.tf_options = tf_options

  def iterate(self):

After Change

    self.header = None

    if compression_type is None:
      compression_type = "GZIP" if input_path.endswith(".gz") else ""

    self.reader = tfrecord_reader.TFRecordReader.from_file(
        input_path, compression_type)
    if self.reader is None:
      raise IOError(errno.EIO,
                    "Error trying to open %s for reading" % input_path)

  def iterate(self):
    Returns an iterator for going through all the file"s records.
    while self.reader.get_next():
      yield self.proto.FromString(self.reader.get_record())

Instances