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 { ///