diff --git a/test/Microsoft.Extensions.ML.Tests/FileLoaderTests.cs b/test/Microsoft.Extensions.ML.Tests/FileLoaderTests.cs index efed95119d..9f3b5448da 100644 --- a/test/Microsoft.Extensions.ML.Tests/FileLoaderTests.cs +++ b/test/Microsoft.Extensions.ML.Tests/FileLoaderTests.cs @@ -11,6 +11,7 @@ using Microsoft.Extensions.Primitives; using Microsoft.ML.Data; using Microsoft.ML.TestFramework; +using Microsoft.ML.TestFrameworkCommon; using Xunit; using Xunit.Abstractions; @@ -72,7 +73,7 @@ public void can_reload_model() File.WriteAllText("testdata.txt", "test"); - Assert.True(changed.WaitOne(2000), "FileLoader ChangeToken didn't fire before the allotted time."); + Assert.True(changed.WaitOne(AsyncTestHelper.UnexpectedTimeout), "FileLoader ChangeToken didn't fire before the allotted time."); }