diff --git a/evaluation/utils/shared.py b/evaluation/utils/shared.py index 972698659b..e769c9e705 100644 --- a/evaluation/utils/shared.py +++ b/evaluation/utils/shared.py @@ -267,7 +267,7 @@ def run_evaluation( for _, instance in dataset.iterrows(): instance_queue.put(instance) - total_instances = instance_queue.qsize() + total_instances = len(dataset) pbar = tqdm(total=total_instances, desc='Instances processed') output_fp = open(output_file, 'a')