From d3181f0bc17369c71bb11578ffb0014ce7de01c7 Mon Sep 17 00:00:00 2001 From: Mamba413 Date: Wed, 24 Apr 2024 09:38:29 +0800 Subject: [PATCH] Update test_alg.py --- python/pytest/test_alg.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/pytest/test_alg.py b/python/pytest/test_alg.py index 6155b760..18d6b701 100644 --- a/python/pytest/test_alg.py +++ b/python/pytest/test_alg.py @@ -134,7 +134,8 @@ def test_binomial(): test_data = load_data("binomial_test") def assert_reg(coef, fit_intercept=True, rel=0.01, abs=0.01): - if sys.version_info[0] + 0.1 * sys.version_info[1] < 3.6: + sys_version_num = sys.version_info[0] + 0.1 * sys.version_info[1] + if sys_version_num < 3.6 or sys_version_num >= 3.9: return nonzero = np.nonzero(coef)[0] new_x = data.x[:, nonzero]