import or skip catboost

rapidsai · Dec 11, 2024 · 0b4f5fa · 0b4f5fa
1 parent 5b11e66
commit 0b4f5fa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/ci/cudf_pandas_scripts/third-party-integration/run-library-tests.sh b/ci/cudf_pandas_scripts/third-party-integration/run-library-tests.sh
@@ -20,7 +20,6 @@ runtest() {
 
     pytest \
     $plugin \
-    --ignore=${TEST_DIR}/test_catboost.py \
     -v \
     --continue-on-collection-errors \
     --cache-clear \
@@ -40,7 +39,6 @@ main() {
     pytest \
     --compare \
     -p cudf.pandas \
-    --ignore=${TEST_DIR}/test_catboost.py \
     -v \
     --continue-on-collection-errors \
     --cache-clear \

diff --git a/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_catboost.py b/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_catboost.py
@@ -3,7 +3,13 @@
 import numpy as np
 import pandas as pd
 import pytest
-from catboost import CatBoostClassifier, CatBoostRegressor, Pool
+
+try:
+    from catboost import CatBoostClassifier, CatBoostRegressor, Pool
+except Exception:
+    pytest.skip(
+        "ValueError: numpy.dtype size changed, may indicate binary incompatibility. Expected 96 from C header, got 88 from PyObject"
+    )
 from sklearn.datasets import make_classification, make_regression
 
 rng = np.random.default_rng(seed=42)