Addressed PR comments

EducationalTestingService · Oct 12, 2022 · f1db9a9 · f1db9a9
1 parent 1b60b1b
commit f1db9a9
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/regression_tests/test_gector_roberta.py b/regression_tests/test_gector_roberta.py
@@ -65,7 +65,6 @@ def predict_for_file(input_file, temp_file, model, batch_size=32):
         if len(batch) == batch_size:
             preds, cnt = model.handle_batch(batch)
             predictions.extend(preds)
-            cnt_corrections += cnt
             batch = []
     if batch:
         preds, cnt = model.handle_batch(batch)