Support categorical split in tree model dump. (#7036)

2021-06-18 16:46:20 +08:00
parent 7968c0d051
commit 29f8fd6fee
8 changed files with 263 additions and 46 deletions
--- a/tests/python-gpu/test_gpu_updaters.py
+++ b/tests/python-gpu/test_gpu_updaters.py
@@ -71,7 +71,6 @@ class TestGPUUpdaters:
    @settings(deadline=None)
    @pytest.mark.skipif(**tm.no_pandas())
    def test_categorical(self, rows, cols, rounds, cats):
-        pytest.xfail(reason='TestGPUUpdaters::test_categorical is flaky')
        self.run_categorical_basic(rows, cols, rounds, cats)

    def test_categorical_32_cat(self):