Merge pull request #1232 from billvsme/fix/category_index_error

fix DecoderOnlyEmbedderICLSameDatasetTrainDataset category index error

Author: hanhainebula

FlagEmbedding/finetune/embedder/decoder_only/icl/dataset.py

@@ -113,8 +113,8 @@ def _create_batch_data(self, batch_raw_data):
                         add_special_tokens=False,
                     )['input_ids']
                 )
-                for i in range(len(tmp_passages)):
-                    tmp_passages[i] += self.args.icl_suffix_str
+                for j in range(len(tmp_passages)):
+                    tmp_passages[j] += self.args.icl_suffix_str
             else:
                 if self.args.passage_instruction_for_retrieval is not None:
                     tmp_passages = [