Broken - E2E tests being updated.

sillsdev · Nov 6, 2024 · 089bb5b · 089bb5b
1 parent 6987a7a
commit 089bb5b
Show file tree

Hide file tree

Showing 6 changed files with 285 additions and 116 deletions.
diff --git a/src/Machine/src/Serval.Machine.Shared/Configuration/IMachineBuilderExtensions.cs b/src/Machine/src/Serval.Machine.Shared/Configuration/IMachineBuilderExtensions.cs
@@ -196,6 +196,7 @@ public static IMachineBuilder AddMemoryDataAccess(this IMachineBuilder builder)
         builder.Services.AddMemoryDataAccess(o =>
         {
             o.AddRepository<TranslationEngine>();
+            o.AddRepository<WordAlignmentEngine>();
             o.AddRepository<RWLock>();
             o.AddRepository<TrainSegmentPair>();
             o.AddRepository<OutboxMessage>();
@@ -232,6 +233,23 @@ await c.Indexes.CreateOrUpdateAsync(
                         );
                     }
                 );
+                o.AddRepository<WordAlignmentEngine>(
+                    "word_alignment_engines",
+                    mapSetup: m => m.SetIgnoreExtraElements(true),
+                    init: async c =>
+                    {
+                        await c.Indexes.CreateOrUpdateAsync(
+                            new CreateIndexModel<WordAlignmentEngine>(
+                                Builders<WordAlignmentEngine>.IndexKeys.Ascending(e => e.EngineId)
+                            )
+                        );
+                        await c.Indexes.CreateOrUpdateAsync(
+                            new CreateIndexModel<WordAlignmentEngine>(
+                                Builders<WordAlignmentEngine>.IndexKeys.Ascending(e => e.CurrentBuild!.BuildJobRunner)
+                            )
+                        );
+                    }
+                );
                 o.AddRepository<RWLock>("locks");
                 o.AddRepository<TrainSegmentPair>(
                     "train_segment_pairs",

diff --git a/src/Machine/src/Serval.Machine.Shared/Services/ClearMLMonitorService.cs b/src/Machine/src/Serval.Machine.Shared/Services/ClearMLMonitorService.cs
@@ -48,15 +48,15 @@ private async Task MonitorClearMLTasksPerDomain(IServiceScope scope, Cancellatio
         try
         {
             var translationBuildJobService = scope.ServiceProvider.GetRequiredService<
-                IBuildJobService<ITrainingEngine>
+                IBuildJobService<TranslationEngine>
             >();
             var wordAlignmentBuildJobService = scope.ServiceProvider.GetRequiredService<
                 IBuildJobService<WordAlignmentEngine>
             >();
 
             Dictionary<ITrainingEngine, IBuildJobService<ITrainingEngine>> engineToBuildServiceDict = (
                 await translationBuildJobService.GetBuildingEnginesAsync(BuildJobRunnerType.ClearML, cancellationToken)
-            ).ToDictionary(e => e, e => translationBuildJobService);
+            ).ToDictionary(e => (ITrainingEngine)e, e => (IBuildJobService<ITrainingEngine>)translationBuildJobService);
 
             foreach (
                 var engine in await wordAlignmentBuildJobService.GetBuildingEnginesAsync(