Skip to content

Commit 2617168

Browse files
authored
Skip catboost test on Python 3.14 (#593)
1 parent a11c7ce commit 2617168

2 files changed

Lines changed: 25 additions & 5 deletions

File tree

.github/actions/setup/action.yml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,3 +47,12 @@ runs:
4747
python -c "import torch; print('PyTorch:', torch.__version__)"
4848
python -c "import torch; print('CUDA:', torch.version.cuda)"
4949
shell: bash
50+
51+
# TODO: Include catboost in Python 3.14 CI when catboost supports it:
52+
# https://github.com/catboost/catboost/issues/2943
53+
- name: Exclude catboost on Python 3.14 from pyproject.toml
54+
if: ${{ inputs.python-version == '3.14' }}
55+
run: |
56+
sed -i '/ "catboost",/d' pyproject.toml
57+
cat pyproject.toml
58+
shell: bash

test/gbdt/test_gbdt.py

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import os.path as osp
2+
import sys
23
import tempfile
34

45
import pytest
@@ -12,11 +13,21 @@
1213
from torch_frame.testing.text_embedder import HashTextEmbedder
1314

1415

15-
@pytest.mark.parametrize('gbdt_cls', [
16-
CatBoost,
17-
XGBoost,
18-
LightGBM,
19-
])
16+
@pytest.mark.parametrize(
17+
'gbdt_cls',
18+
[
19+
# TODO: Run CatBoost test on Python 3.14 once supported
20+
# https://github.com/catboost/catboost/issues/2943
21+
pytest.param(
22+
CatBoost,
23+
marks=pytest.mark.skipif(
24+
sys.version_info >= (3, 14),
25+
reason="Not supported on Python 3.14",
26+
),
27+
),
28+
XGBoost,
29+
LightGBM,
30+
])
2031
@pytest.mark.parametrize('stypes', [
2132
[stype.numerical],
2233
[stype.categorical],

0 commit comments

Comments
 (0)