Skip to content

Commit 3a1c8fd

Browse files
authored
Merge pull request #7 from EngineeringSoftware/fix-model
Fix model
2 parents 5d37a96 + f304402 commit 3a1c8fd

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

roosterize/ml/naming/MultiSourceSeq2Seq.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,9 @@ def process_data_impl(
349349
# Inputs
350350
all_inputs: Dict[str, List[List[str]]] = self.get_all_inputs(lemmas, docs_sub_tokenizers)
351351
for input_type, src_sentences in all_inputs.items():
352-
IOUtils.dump(output_processed_data_dir/f"src.{input_type}.txt", src_sentences, IOUtils.Format.txtList)
352+
IOUtils.dump(output_processed_data_dir/f"src.{input_type}.txt",
353+
"".join([" ".join(sent) + "\n" for sent in src_sentences]),
354+
IOUtils.Format.txt)
353355

354356
# Outputs
355357
IOUtils.dump(

0 commit comments

Comments
 (0)