Support more input types for categorical data. (#7220)

* Support more input types for categorical data.

* Shorten the type name from "categorical" to "c".
* Tests for np/cp array and scipy csr/csc/coo.
* Specify the type for feature info.

tests/python-gpu/test_from_cudf.py

@@ -172,7 +172,7 @@ Arrow specification.'''
         _test_cudf_metainfo(xgb.DeviceQuantileDMatrix)
 
     @pytest.mark.skipif(**tm.no_cudf())
-    def test_categorical(self):
+    def test_cudf_categorical(self):
         import cudf
         _X, _y = tm.make_categorical(100, 30, 17, False)
         X = cudf.from_pandas(_X)
@@ -180,11 +180,11 @@ Arrow specification.'''
 
         Xy = xgb.DMatrix(X, y, enable_categorical=True)
         assert len(Xy.feature_types) == X.shape[1]
-        assert all(t == "categorical" for t in Xy.feature_types)
+        assert all(t == "c" for t in Xy.feature_types)
 
         Xy = xgb.DeviceQuantileDMatrix(X, y, enable_categorical=True)
         assert len(Xy.feature_types) == X.shape[1]
-        assert all(t == "categorical" for t in Xy.feature_types)
+        assert all(t == "c" for t in Xy.feature_types)
 
 
 @pytest.mark.skipif(**tm.no_cudf())

tests/python-gpu/test_from_cupy.py

@@ -169,6 +169,19 @@ Arrow specification.'''
         X = cp.random.random((n, 2))
         xgb.DMatrix(X.toDlpack())
 
+    @pytest.mark.skipif(**tm.no_cupy())
+    def test_cupy_categorical(self):
+        import cupy as cp
+        n_features = 10
+        X, y = tm.make_categorical(10, n_features, n_categories=4, onehot=False)
+        X = cp.asarray(X.values.astype(cp.float32))
+        y = cp.array(y)
+        feature_types = ['c'] * n_features
+
+        assert isinstance(X, cp.ndarray)
+        Xy = xgb.DMatrix(X, y, feature_types=feature_types)
+        np.testing.assert_equal(np.array(Xy.feature_types), np.array(feature_types))
+
     @pytest.mark.skipif(**tm.no_cupy())
     def test_dlpack_device_dmat(self):
         import cupy as cp