Update AutoMLExperimentTests.cs to fix timeout error (#6638)

dotnet · Apr 30, 2023 · a18b9cb · a18b9cb
1 parent 3386245
commit a18b9cb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/Microsoft.ML.AutoML.Tests/AutoMLExperimentTests.cs b/test/Microsoft.ML.AutoML.Tests/AutoMLExperimentTests.cs
@@ -223,7 +223,7 @@ public async Task AutoMLExperiment_UCI_Adult_Train_Test_Split_Test()
             experiment.SetDataset(context.Data.TrainTestSplit(data))
                     .SetBinaryClassificationMetric(BinaryClassificationMetric.AreaUnderRocCurve, DatasetUtil.UciAdultLabel)
                     .SetPipeline(pipeline)
-                    .SetTrainingTimeInSeconds(1);
+                    .SetMaxModelToExplore(1);
 
             var result = await experiment.RunAsync();
             result.Metric.Should().BeGreaterThan(0.8);
@@ -248,7 +248,7 @@ public async Task AutoMLExperiment_UCI_Adult_CV_5_Test()
             experiment.SetDataset(data, 5)
                     .SetBinaryClassificationMetric(BinaryClassificationMetric.AreaUnderRocCurve, DatasetUtil.UciAdultLabel)
                     .SetPipeline(pipeline)
-                    .SetTrainingTimeInSeconds(10);
+                    .SetMaxModelToExplore(1);
 
             var result = await experiment.RunAsync();
             result.Metric.Should().BeGreaterThan(0.8);
@@ -275,7 +275,7 @@ public async Task AutoMLExperiment_Iris_CV_5_Test()
             experiment.SetDataset(data, 5)
                     .SetMulticlassClassificationMetric(MulticlassClassificationMetric.MacroAccuracy, label)
                     .SetPipeline(pipeline)
-                    .SetTrainingTimeInSeconds(10);
+                    .SetMaxModelToExplore(1);
 
             var result = await experiment.RunAsync();
             result.Metric.Should().BeGreaterThan(0.8);
@@ -302,7 +302,7 @@ public async Task AutoMLExperiment_Iris_Train_Test_Split_Test()
             experiment.SetDataset(context.Data.TrainTestSplit(data))
                     .SetMulticlassClassificationMetric(MulticlassClassificationMetric.MacroAccuracy, label)
                     .SetPipeline(pipeline)
-                    .SetTrainingTimeInSeconds(10);
+                    .SetMaxModelToExplore(1);
 
             var result = await experiment.RunAsync();
             result.Metric.Should().BeGreaterThan(0.8);