RunSpecific in test

jwood803 · Dec 17, 2020 · fed12b2 · fed12b2
1 parent f9f328d
commit fed12b2
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/test/Microsoft.ML.AutoML.Tests/AutoFitTests.cs b/test/Microsoft.ML.AutoML.Tests/AutoFitTests.cs
@@ -442,9 +442,11 @@ public void AutoFitWithPresplittedData()
 
         }
 
-        [LightGBMFact]
-        public void AutoFitMaxExperimentTimeTest()
+        [Theory, IterationData(10)]
+        [TestCategory("RunSpecificTest")]
+        public void AutoFitMaxExperimentTimeTest(int iteration)
         {
+            Console.WriteLine(iteration);
             // A single binary classification experiment takes less than 5 seconds.
             // System.OperationCanceledException is thrown when ongoing experiment
             // is canceled and at least one model has been generated.
@@ -455,8 +457,12 @@ public void AutoFitMaxExperimentTimeTest()
             var columnInference = context.Auto().InferColumns(dataPath, DatasetUtil.UciAdultLabel);
             var textLoader = context.Data.CreateTextLoader(columnInference.TextLoaderOptions);
             var trainData = textLoader.Load(dataPath);
+
+            var experSettings = new BinaryExperimentSettings { MaxExperimentTimeInSeconds = 60};
+            experSettings.Trainers.Remove(BinaryClassificationTrainer.LightGbm);
+
             var experiment = context.Auto()
-                .CreateBinaryClassificationExperiment(15)
+                .CreateBinaryClassificationExperiment(experSettings)
                 .Execute(trainData, new ColumnInformation() { LabelColumnName = DatasetUtil.UciAdultLabel });
 
             // Ensure the (last) model that was training when maximum experiment time was reached has been stopped,