diff --git a/test/Microsoft.ML.OnnxTransformerTest/DnnImageFeaturizerTest.cs b/test/Microsoft.ML.OnnxTransformerTest/DnnImageFeaturizerTest.cs index b9f68944c9..9e9442e7fa 100644 --- a/test/Microsoft.ML.OnnxTransformerTest/DnnImageFeaturizerTest.cs +++ b/test/Microsoft.ML.OnnxTransformerTest/DnnImageFeaturizerTest.cs @@ -54,10 +54,14 @@ public DnnImageFeaturizerTests(ITestOutputHelper helper) : base(helper) } [OnnxFact] - //Skipping test temporarily. This test will be re-enabled once the cause of failures has been determined - [Trait("Category", "SkipInCI")] public void TestDnnImageFeaturizer() { + //skip running for x86 as this test using too much memory (over 2GB limit on x86) + //and very like to hit memory related issue when running on CI + //TODO: optimized memory usage in related code and enable x86 test run + if (!Environment.Is64BitProcess) + return; + var samplevector = GetSampleArrayData(); var dataView = DataViewConstructionUtils.CreateFromList(Env, @@ -125,12 +129,15 @@ public void OnnxFeaturizerWorkout() } } - // Onnx is only supported on x64 Windows [OnnxFact] - //Skipping test temporarily. This test will be re-enabled once the cause of failures has been determined - [Trait("Category", "SkipInCI")] public void TestOldSavingAndLoading() { + //skip running for x86 as this test using too much memory (over 2GB limit on x86) + //and very like to hit memory related issue when running on CI + //TODO: optimized memory usage in related code and enable x86 run + if (!Environment.Is64BitProcess) + return; + var samplevector = GetSampleArrayData(); var dataView = ML.Data.LoadFromEnumerable(