From 3e040c06316e8fa7af093b54891fafa8d3d97934 Mon Sep 17 00:00:00 2001
From: Peter Wang <peter.wang9812@gmail.com>
Date: Tue, 12 Dec 2023 04:08:18 -0600
Subject: [PATCH] Change typo in matrix

---
 xgboost_ray/matrix.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/xgboost_ray/matrix.py b/xgboost_ray/matrix.py
index f3f7523..71f4f81 100644
--- a/xgboost_ray/matrix.py
+++ b/xgboost_ray/matrix.py
@@ -49,11 +49,11 @@ class RayDataset:
     LEGACY_MATRIX = True
 
 try:
-    from xgboost.core import QuantileDmatrix
+    from xgboost.core import QuantileDMatrix
 
     QUANTILE_AVAILABLE = True
 except ImportError:
-    QuantileDmatrix = object
+    QuantileDMatrix = object
     QUANTILE_AVAILABLE = False
 
 if TYPE_CHECKING: