Fix tests

sillsdev · Apr 3, 2024 · de20af2 · de20af2
1 parent dce1ae2
commit de20af2
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/tests/Serval.E2ETests/ServalApiTests.cs b/tests/Serval.E2ETests/ServalApiTests.cs
@@ -125,8 +125,8 @@ public async Task NmtBatch()
     [Test]
     public async Task NmtQueueMultiple()
     {
-        const int NUM_ENGINES = 9;
-        const int NUM_WORKERS = 1;
+        const int NUM_ENGINES = 10;
+        const int NUM_WORKERS = 8;
         string[] engineIds = new string[NUM_ENGINES];
         for (int i = 0; i < NUM_ENGINES; i++)
         {

diff --git a/tests/Serval.E2ETests/ServalClientHelper.cs b/tests/Serval.E2ETests/ServalClientHelper.cs
@@ -69,7 +69,16 @@ public TranslationBuildConfig InitTranslationBuildConfig()
         {
             Pretranslate = [],
             TrainOn = null,
-            Options = "{\"max_steps\":10, \"train_params\": {\"per_device_train_batch_size\":4}}"
+            Options = """
+                {
+                    "max_steps": 10,
+                    "parent_model_name": "hf-internal-testing/tiny-random-nllb",
+                    "train_params":
+                    {
+                        "per_device_train_batch_size": 4
+                    }
+                }
+                """
         };
         return TranslationBuildConfig;
     }