506a594e8e5fd755bc02ea2c5fe6437b8b682057,fonduer/utils/udf.py,UDF,run,#UDF#,160

Before Change

        while True:
            try:
                x = self.in_queue.get(True, QUEUE_TIMEOUT)
                for y in self.apply(x, **self.apply_kwargs):

                    // If an out_queue is provided, add to that, else add to session
                    if self.out_queue is not None:
                        self.out_queue.put(y, True, QUEUE_TIMEOUT)
                    else:
                        self.session.add(y)
                self.in_queue.task_done()
            except Empty:
                break
        self.session.commit()

After Change

        while True:
            try:
                x = self.in_queue.get(True, QUEUE_TIMEOUT)
                self.session.add_all(y for y in self.apply(x, **self.apply_kwargs))
                self.in_queue.task_done()
            except Empty:
                break

Instances