Fix scal failing after one doc

hical · Apr 17, 2020 · 73034b0 · 73034b0
1 parent 8d6dfab
commit 73034b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CALEngine/src/bmi_para_scal.cc b/CALEngine/src/bmi_para_scal.cc
@@ -60,7 +60,7 @@ void BMI_para_scal::record_judgment_batch(vector<pair<string, int>> _judgments){
         shuffle(batch.begin(), batch.end(), rand_generator);
         for(int i = 0; i < batch.size(); i++){
             if(selector[i]) judgment_queue.push_back(batch[i]);
-            judgments[paragraphs->translate_index(batch[i])] = -2;
+            else judgments[paragraphs->translate_index(batch[i])] = -2;
         }
         B = B + ceil(B/10.0);
     }