From b9e48ef81116e6c3697c39fa488d855fc7dfec95 Mon Sep 17 00:00:00 2001 From: Michael Sharp Date: Fri, 19 Jan 2024 12:34:40 -0700 Subject: [PATCH] testing light gbm tests sequentially --- test/Microsoft.ML.TestFramework/BaseTestClass.cs | 2 +- test/Microsoft.ML.Tests/TrainerEstimators/TreeEstimators.cs | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/test/Microsoft.ML.TestFramework/BaseTestClass.cs b/test/Microsoft.ML.TestFramework/BaseTestClass.cs index 102fa205ed..6fc4350eea 100644 --- a/test/Microsoft.ML.TestFramework/BaseTestClass.cs +++ b/test/Microsoft.ML.TestFramework/BaseTestClass.cs @@ -81,7 +81,7 @@ void IDisposable.Dispose() Cleanup(); Process proc = Process.GetCurrentProcess(); Console.WriteLine($"Finished test: {FullTestName} " + - $"with memory usage {proc.WorkingSet64.ToString("N", CultureInfo.InvariantCulture)}"); + $"with memory usage {proc.WorkingSet64.ToString("N", CultureInfo.InvariantCulture)} and max memory usage {proc.PeakWorkingSet64.ToString("N", CultureInfo.InvariantCulture)}"); } protected virtual void Initialize() diff --git a/test/Microsoft.ML.Tests/TrainerEstimators/TreeEstimators.cs b/test/Microsoft.ML.Tests/TrainerEstimators/TreeEstimators.cs index 42540f478f..1794cce49f 100644 --- a/test/Microsoft.ML.Tests/TrainerEstimators/TreeEstimators.cs +++ b/test/Microsoft.ML.Tests/TrainerEstimators/TreeEstimators.cs @@ -24,6 +24,10 @@ namespace Microsoft.ML.Tests.TrainerEstimators { + [CollectionDefinition(nameof(NoParallelizationDefinition), DisableParallelization = true)] + public class NoParallelizationDefinition { } + + [Collection(nameof(NoParallelizationDefinition))] public partial class TrainerEstimators : TestDataPipeBase { ///