Update test_gbt_single.py

NVlabs · Jan 23, 2025 · bfe3973 · bfe3973
1 parent 19bbdaa
commit bfe3973
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_gbt_single.py b/tests/test_gbt_single.py
@@ -273,7 +273,7 @@ def test_l2_cpu(self):
         model._model.reset()
         model.set_bias_from_targets(y)
         train_loss = model.fit(X, y, self.n_epochs)
-        value = 2.0
+        value = 3.0
         self.assertTrue(train_loss < value, f'Expected loss = {train_loss} < {value}')
         X_categorical, y_categorical = self.cat_data
         model._model.reset()