Skip to content

Commit c2685ea

Browse files
committed
Revert "RunSpecific in test"
This reverts commit fed12b2.
1 parent 79407db commit c2685ea

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

test/Microsoft.ML.AutoML.Tests/AutoFitTests.cs

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -442,11 +442,9 @@ public void AutoFitWithPresplittedData()
442442

443443
}
444444

445-
[Theory, IterationData(10)]
446-
[TestCategory("RunSpecificTest")]
447-
public void AutoFitMaxExperimentTimeTest(int iteration)
445+
[LightGBMFact]
446+
public void AutoFitMaxExperimentTimeTest()
448447
{
449-
Console.WriteLine(iteration);
450448
// A single binary classification experiment takes less than 5 seconds.
451449
// System.OperationCanceledException is thrown when ongoing experiment
452450
// is canceled and at least one model has been generated.
@@ -457,12 +455,8 @@ public void AutoFitMaxExperimentTimeTest(int iteration)
457455
var columnInference = context.Auto().InferColumns(dataPath, DatasetUtil.UciAdultLabel);
458456
var textLoader = context.Data.CreateTextLoader(columnInference.TextLoaderOptions);
459457
var trainData = textLoader.Load(dataPath);
460-
461-
var experSettings = new BinaryExperimentSettings { MaxExperimentTimeInSeconds = 60};
462-
experSettings.Trainers.Remove(BinaryClassificationTrainer.LightGbm);
463-
464458
var experiment = context.Auto()
465-
.CreateBinaryClassificationExperiment(experSettings)
459+
.CreateBinaryClassificationExperiment(15)
466460
.Execute(trainData, new ColumnInformation() { LabelColumnName = DatasetUtil.UciAdultLabel });
467461

468462
// Ensure the (last) model that was training when maximum experiment time was reached has been stopped,

0 commit comments

Comments
 (0)