def collect(self):
    pdf = self._yield_pdf()
    return [Row(pdf, self.schema, i, self._name_to_column_map) for i in range(0, len(pdf))]
def collect(self):
    pdf = self._yield_pdf()
    return [Row(pdf, self.schema, i, self._name_to_column_map) for i in range(0, len(pdf))]