From 6736525e9ab5e4de9156a26b6e3fd523130e06a2 Mon Sep 17 00:00:00 2001
From: Enkidu93 <lowryec17@gcc.edu>
Date: Mon, 4 Dec 2023 18:17:59 -0500
Subject: [PATCH] Fix key term ids logic

---
 src/SIL.Machine.AspNetCore/Services/NmtPreprocessBuildJob.cs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/SIL.Machine.AspNetCore/Services/NmtPreprocessBuildJob.cs b/src/SIL.Machine.AspNetCore/Services/NmtPreprocessBuildJob.cs
index e62dd5e91..79445f328 100644
--- 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);
                 }