Skip to content

Commit 37c523b

Browse files
Support CatBoost in Python 3.13 (#523)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 1758a40 commit 37c523b

File tree

2 files changed

+5
-25
lines changed

2 files changed

+5
-25
lines changed

.github/actions/setup/action.yml

-9
Original file line numberDiff line numberDiff line change
@@ -47,12 +47,3 @@ 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.13 CI when catboost supports it:
52-
# https://github.com/catboost/catboost/issues/2748
53-
- name: Exclude catboost on Python 3.13 from pyproject.toml
54-
if: ${{ inputs.python-version == '3.13' }}
55-
run: |
56-
sed -i '/ "catboost",/d' pyproject.toml
57-
cat pyproject.toml
58-
shell: bash

test/gbdt/test_gbdt.py

+5-16
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import os.path as osp
2-
import sys
32
import tempfile
43

54
import pytest
@@ -13,21 +12,11 @@
1312
from torch_frame.testing.text_embedder import HashTextEmbedder
1413

1514

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

0 commit comments

Comments
 (0)