Fix key term ids logic

sillsdev · Dec 4, 2023 · 6736525 · 6736525
1 parent 64a7b05
commit 6736525
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/SIL.Machine.AspNetCore/Services/NmtPreprocessBuildJob.cs b/src/SIL.Machine.AspNetCore/Services/NmtPreprocessBuildJob.cs
@@ -98,7 +98,9 @@ async IAsyncEnumerable<Pretranslation> ProcessRowsAsync()
                     IParallelTextCorpus parallelKeyTermsCorpus = sourceCorpora[CorpusType.Term].AlignRows(
                         targetCorpora[CorpusType.Term]
                     );
-                    corpus.TrainOnTextIds.Add(parallelKeyTermsCorpus.Select(r => r.TextId).Distinct().First()); //Should only be one textId
+                    IEnumerable<string> keyTermsTextIds = parallelKeyTermsCorpus.Select(r => r.TextId).Distinct();
+                    if (keyTermsTextIds.Count() == 1)
+                        corpus.TrainOnTextIds.Add(keyTermsTextIds.First()); //Should only be one textId for key terms
                     parallelCorpora.Add(parallelKeyTermsCorpus);
                 }