Peter's fix

sillsdev · Oct 29, 2024 · f620097 · f620097
1 parent 2c0dd11
commit f620097
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/src/Serval/src/Serval.Translation/Services/EngineService.cs b/src/Serval/src/Serval.Translation/Services/EngineService.cs
@@ -824,7 +824,7 @@ pretranslateFilter is not null
             && referenceFileLocation is not null
         )
         {
-            GetChapters(referenceFileLocation, pretranslateFilter.ScriptureRange)
+            pretranslateChapters = GetChapters(referenceFileLocation, pretranslateFilter.ScriptureRange)
                 .Select(
                     (kvp) =>
                     {

diff --git a/src/Serval/test/Serval.Translation.Tests/Services/EngineServiceTests.cs b/src/Serval/test/Serval.Translation.Tests/Services/EngineServiceTests.cs
@@ -1306,6 +1306,17 @@ await env.Service.StartBuildAsync(
                             new() { CorpusRef = "parallel-corpus1-target1", ScriptureRange = "MAT 1;MRK" }
                         }
                     }
+                ],
+                Pretranslate =
+                [
+                    new PretranslateCorpus
+                    {
+                        ParallelCorpusRef = "parallel-corpus1",
+                        SourceFilters = new List<ParallelCorpusFilter>()
+                        {
+                            new() { CorpusRef = "parallel-corpus1-source1", ScriptureRange = "MAT 2" }
+                        }
+                    }
                 ]
             }
         );
@@ -1340,6 +1351,13 @@ await env.Service.StartBuildAsync(
                                                 new ScriptureChapters { Chapters = { } }
                                             }
                                         },
+                                        PretranslateChapters =
+                                        {
+                                            {
+                                                "MAT",
+                                                new ScriptureChapters { Chapters = { 2 } }
+                                            }
+                                        },
                                         Files =
                                         {
                                             new V1.CorpusFile
@@ -1349,7 +1367,7 @@ await env.Service.StartBuildAsync(
                                                 TextId = "file1.zip"
                                             }
                                         },
-                                        PretranslateAll = true,
+                                        PretranslateAll = false,
                                         TrainOnAll = false
                                     },
                                     new()
@@ -1365,7 +1383,7 @@ await env.Service.StartBuildAsync(
                                                 TextId = "file3.zip"
                                             }
                                         },
-                                        PretranslateAll = true,
+                                        PretranslateAll = false,
                                         TrainOnAll = false
                                     }
                                 }