Internal change

PiperOrigin-RevId: 445359947
Yuze-Liao · Apr 29, 2022 · 298d302 · 298d302
1 parent 2e333cd
commit 298d302
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yggdrasil_decision_forests/learner/random_forest/random_forest_test.cc b/yggdrasil_decision_forests/learner/random_forest/random_forest_test.cc
@@ -473,10 +473,10 @@ TEST_F(RandomForestOnAdult, MaximumSize) {
   train_config_.set_maximum_model_size_in_memory_in_bytes(max_size);
 
   TrainAndEvaluateModel();
-  // Add an extra 2kB to help with the test flakiness.
+  // Add an extra 3kB to help with the test flakiness.
   // Note: the model can be slighly larger than the
   // "set_maximum_model_size_in_memory_in_bytes" directive.
-  EXPECT_LT(model_->ModelSizeInBytes().value(), max_size + 2 * 1024);
+  EXPECT_LT(model_->ModelSizeInBytes().value(), max_size + 3 * 1024);
 
   EXPECT_GT(metric::Accuracy(evaluation_), 0.840);
 }