Validate weights are positive values. (#6115)

2020-09-15 09:03:55 +08:00
parent c6f2b8c841
commit b5f52f0b1b
6 changed files with 15 additions and 9 deletions
--- a/tests/python/test_with_modin.py
+++ b/tests/python/test_with_modin.py
@@ -135,7 +135,7 @@ class TestModin(unittest.TestCase):

        X = np.random.randn(kRows, kCols)
        y = np.random.randn(kRows)
-        w = np.random.randn(kRows).astype(np.float32)
+        w = np.random.uniform(size=kRows).astype(np.float32)
        w_pd = md.DataFrame(w)
        data = xgb.DMatrix(X, y, w_pd)