Temp fix for the race condition during the tests. (dotnet#7021)

* temp fixing the race condition during the tests * making db tests sequential
ibrahim324 · Feb 21, 2024 · 0d2fd60 · 0d2fd60
1 parent 4d4f7dc
commit 0d2fd60
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/test/Microsoft.ML.Tests/DatabaseLoaderTests.cs b/test/Microsoft.ML.Tests/DatabaseLoaderTests.cs
@@ -20,6 +20,9 @@
 
 namespace Microsoft.ML.Tests
 {
+    [CollectionDefinition(nameof(NoParallelizationDefinition), DisableParallelization = true)]
+    public class NoParallelizationDefinition { }
+    [Collection(nameof(NoParallelizationDefinition))]
     public class DatabaseLoaderTests : BaseTestClass
     {
         public DatabaseLoaderTests(ITestOutputHelper output)

diff --git a/test/Microsoft.ML.Tests/Transformers/WordEmbeddingsTests.cs b/test/Microsoft.ML.Tests/Transformers/WordEmbeddingsTests.cs
@@ -12,6 +12,9 @@
 
 namespace Microsoft.ML.Tests.Transformers
 {
+    [CollectionDefinition(nameof(NoParallelizationDefinition), DisableParallelization = true)]
+    public class NoParallelizationDefinition { }
+    [Collection(nameof(NoParallelizationDefinition))]
     public sealed class WordEmbeddingsTests : TestDataPipeBase
     {
         public WordEmbeddingsTests(ITestOutputHelper helper)