Update tests/python_package_test/test_engine.py

microsoft · Apr 20, 2024 · 178ea94 · 178ea94
1 parent b6835eb
commit 178ea94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python_package_test/test_engine.py b/tests/python_package_test/test_engine.py
@@ -966,7 +966,7 @@ def test_early_stopping_is_not_enabled_for_non_positive_stopping_rounds(early_st
         assert "early_stopping_round" not in gbm.params
         assert gbm.num_trees() == num_trees
     elif early_stopping_round == "None":
-        with pytest.raises(TypeError, match="early_stopping_round should be an integer. Got str"):
+        with pytest.raises(TypeError, match="early_stopping_round should be an integer. Got 'str'"):
             gbm = lgb.train(
                 params,
                 lgb_train,