Fix.

dmlc · Jan 8, 2024 · 7b1d550 · 7b1d550
1 parent 6b2c19c
commit 7b1d550
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_distributed/test_with_dask/test_with_dask.py b/tests/test_distributed/test_with_dask/test_with_dask.py
@@ -2251,7 +2251,10 @@ def test_callback(self, client: "Client") -> None:
             )
             for i in range(1, 10):
                 assert os.path.exists(
-                    os.path.join(tmpdir, f"model_{i}.{xgb.callback.TrainingCheckPoint}")
+                    os.path.join(
+                        tmpdir,
+                        f"model_{i}.{xgb.callback.TrainingCheckPoint.default_format}",
+                    )
                 )