Skip to content

Commit 823f44b

Browse files
authored
[ML] Fix flakey progress monitoring test (#1228)
Backport #1226.
1 parent 6267d42 commit 823f44b

File tree

2 files changed

+14
-9
lines changed

2 files changed

+14
-9
lines changed

lib/api/unittest/CDataFrameAnalyzerOutlierTest.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -576,7 +576,8 @@ BOOST_AUTO_TEST_CASE(testProgress) {
576576
LOG_DEBUG(<< sb.GetString());
577577
if (result["phase_progress"]["phase"] == maths::COutliers::COMPUTING_OUTLIERS) {
578578
computingOutliersProgress =
579-
result["phase_progress"]["progress_percent"].GetInt();
579+
std::max(computingOutliersProgress,
580+
result["phase_progress"]["progress_percent"].GetInt());
580581
}
581582
}
582583
}

lib/api/unittest/CDataFrameAnalyzerTrainingTest.cc

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -963,13 +963,15 @@ BOOST_AUTO_TEST_CASE(testProgress) {
963963
std::string phase{result["phase_progress"]["phase"].GetString()};
964964
int progress{result["phase_progress"]["progress_percent"].GetInt()};
965965
if (phase == maths::CBoostedTreeFactory::FEATURE_SELECTION) {
966-
featureSelectionLastProgress = progress;
966+
featureSelectionLastProgress = std::max(featureSelectionLastProgress, progress);
967967
} else if (phase == maths::CBoostedTreeFactory::COARSE_PARAMETER_SEARCH) {
968-
coarseParameterSearchLastProgress = progress;
968+
coarseParameterSearchLastProgress =
969+
std::max(coarseParameterSearchLastProgress, progress);
969970
} else if (phase == maths::CBoostedTreeFactory::FINE_TUNING_PARAMETERS) {
970-
fineTuneParametersLastProgress = progress;
971+
fineTuneParametersLastProgress =
972+
std::max(fineTuneParametersLastProgress, progress);
971973
} else if (phase == maths::CBoostedTreeFactory::FINAL_TRAINING) {
972-
finalTrainLastProgress = progress;
974+
finalTrainLastProgress = std::max(finalTrainLastProgress, progress);
973975
}
974976
}
975977
}
@@ -1052,17 +1054,19 @@ BOOST_AUTO_TEST_CASE(testProgressFromRestart) {
10521054
std::string phase{result["phase_progress"]["phase"].GetString()};
10531055
int progress{result["phase_progress"]["progress_percent"].GetInt()};
10541056
if (phase == maths::CBoostedTreeFactory::FEATURE_SELECTION) {
1055-
featureSelectionLastProgress = progress;
1057+
featureSelectionLastProgress = std::max(featureSelectionLastProgress, progress);
10561058
} else if (phase == maths::CBoostedTreeFactory::COARSE_PARAMETER_SEARCH) {
1057-
coarseParameterSearchLastProgress = progress;
1059+
coarseParameterSearchLastProgress =
1060+
std::max(coarseParameterSearchLastProgress, progress);
10581061
} else if (phase == maths::CBoostedTreeFactory::FINE_TUNING_PARAMETERS) {
10591062
if (progress > 0) {
10601063
fineTuneParametersFirstProgress =
10611064
std::min(fineTuneParametersFirstProgress, progress);
10621065
}
1063-
fineTuneParametersLastProgress = progress;
1066+
fineTuneParametersLastProgress =
1067+
std::max(fineTuneParametersLastProgress, progress);
10641068
} else if (phase == maths::CBoostedTreeFactory::FINAL_TRAINING) {
1065-
finalTrainLastProgress = progress;
1069+
finalTrainLastProgress = std::max(finalTrainLastProgress, progress);
10661070
}
10671071
}
10681072
}

0 commit comments

Comments
 (0)