Debug

Author: betatim

ci/run_cuml_singlegpu_pytests.sh

@@ -4,4 +4,4 @@
 # Support invoking run_cuml_singlegpu_pytests.sh outside the script directory
 cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/cuml/tests || exit 1
 
-python -m pytest --cache-clear --ignore=dask "$@" .
+python -m pytest --cache-clear -sv --tb=auto --disable-warnings --ignore=dask "$@" .

python/cuml/tests/test_naive_bayes.py

@@ -474,6 +474,10 @@ def test_gaussian_parameters(priors, var_smoothing, nlp_20news):
     X = sparse_scipy_to_cp(X[:nrows], x_dtype).todense()[:, :ncols]
     y = y.astype(y_dtype)[:nrows]
 
+    print()
+    print(f"{np.unique(y)=}")
+    print(f"{y=}")
+
     if priors == "balanced":
         priors = cp.array([1 / 20] * 20)
     elif priors == "unbalanced":