File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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
Original file line number Diff line number Diff line change 11import os .path as osp
2+ import sys
23import tempfile
34
45import pytest
1213from 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 ],
You can’t perform that action at this time.
0 commit comments