Change alignment data structure

sillsdev · Jan 9, 2025 · 6f0b6c5 · 6f0b6c5
1 parent f905c2d
commit 6f0b6c5
Showing 1 changed file with 5 additions and 26 deletions.
diff --git a/src/SIL.Machine/Translation/WordAlignmentResult.cs b/src/SIL.Machine/Translation/WordAlignmentResult.cs
@@ -1,6 +1,6 @@
-using System;
-using System.Collections.Generic;
+using System.Collections.Generic;
 using System.Linq;
+using SIL.Machine.Corpora;
 
 namespace SIL.Machine.Translation
 {
@@ -10,39 +10,18 @@ public WordAlignmentResult(
             IEnumerable<string> sourceTokens,
             IEnumerable<string> targetTokens,
             IEnumerable<double> confidences,
-            WordAlignmentMatrix alignment
+            IEnumerable<AlignedWordPair> alignedWordPairs
         )
         {
             SourceTokens = sourceTokens.ToArray();
             TargetTokens = targetTokens.ToArray();
             Confidences = confidences.ToArray();
-            if (Confidences.Count != TargetTokens.Count)
-            {
-                throw new ArgumentException(
-                    "The confidences must be the same length as the target segment.",
-                    nameof(confidences)
-                );
-            }
-            Alignment = alignment;
-            if (Alignment.RowCount != SourceTokens.Count)
-            {
-                throw new ArgumentException(
-                    "The alignment source length must be the same length as the source segment.",
-                    nameof(alignment)
-                );
-            }
-            if (Alignment.ColumnCount != TargetTokens.Count)
-            {
-                throw new ArgumentException(
-                    "The alignment target length must be the same length as the target segment.",
-                    nameof(alignment)
-                );
-            }
+            AlignedWordPairs = alignedWordPairs;
         }
 
         public IReadOnlyList<string> SourceTokens { get; }
         public IReadOnlyList<string> TargetTokens { get; }
         public IReadOnlyList<double> Confidences { get; }
-        public WordAlignmentMatrix Alignment { get; }
+        public IEnumerable<AlignedWordPair> AlignedWordPairs { get; }
     }
 }