Change mergedtextcorpus parameter

sillsdev · Nov 15, 2024 · 2058d4d · 2058d4d
1 parent 14697e4
commit 2058d4d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 12 deletions.
diff --git a/src/SIL.Machine/Corpora/CorporaExtensions.cs b/src/SIL.Machine/Corpora/CorporaExtensions.cs
@@ -544,20 +544,12 @@ public static ITextCorpus ChooseRandom(
             int seed
         )
         {
-            return new MergedTextCorpus(
-                new NParallelTextCorpus(corpora) { AllRows = allRows.ToArray() },
-                MergeRule.Random,
-                seed
-            );
+            return new MergedTextCorpus(corpora, allRows.ToArray(), MergeRule.Random, seed);
         }
 
         public static ITextCorpus ChooseFirst(this IEnumerable<ITextCorpus> corpora, IEnumerable<bool> allRows)
         {
-            return new MergedTextCorpus(
-                new NParallelTextCorpus(corpora) { AllRows = allRows.ToArray() },
-                MergeRule.First,
-                0
-            );
+            return new MergedTextCorpus(corpora, allRows.ToArray(), MergeRule.First, 0);
         }
 
         #endregion

diff --git a/src/SIL.Machine/Corpora/MergedTextCorpus.cs b/src/SIL.Machine/Corpora/MergedTextCorpus.cs
@@ -13,9 +13,14 @@ public class MergedTextCorpus : TextCorpusBase
 
         private readonly Random _random;
 
-        public MergedTextCorpus(NParallelTextCorpus nParallelTextCorpus, MergeRule mergeRule, int seed)
+        public MergedTextCorpus(
+            IEnumerable<ITextCorpus> corpora,
+            IEnumerable<bool> allRows,
+            MergeRule mergeRule,
+            int seed
+        )
         {
-            _corpus = nParallelTextCorpus;
+            _corpus = new NParallelTextCorpus(corpora) { AllRows = allRows.ToList() };
             _mergeRule = mergeRule;
             _random = new Random(seed);
         }