45e62d2ce97bba98293e3202795d94ecffd38b07,src/fonduer/utils/udf.py,UDFRunner,apply_mt,#UDFRunner#Any#Any#,86

Before Change

        // Fill queue with docs; progress bar will not be updated until this is done
        for x in xs:
            in_queue.put(x)
        in_queue.put(UDF.QUEUE_CLOSED)

        while any([udf.is_alive() for udf in self.udfs]) and count_parsed < total_count:

            y = out_queue.get()

After Change

            if docs_added < total_count:
                in_queue.put(next(xs_generator))
                docs_added += 1
            if docs_added == total_count:
                in_queue.put(UDF.QUEUE_CLOSED)
                docs_added += 1

            y = out_queue.get()

            // Update progress bar whenever an item is processed
            if y == UDF.TASK_DONE:

Instances