diff --git a/.github/actions/setup/action.yml b/.github/actions/setup/action.yml index 1d49e3cee..30e8c0c4a 100644 --- a/.github/actions/setup/action.yml +++ b/.github/actions/setup/action.yml @@ -42,12 +42,3 @@ runs: python -c "import torch; print('PyTorch:', torch.__version__)" python -c "import torch; print('CUDA:', torch.version.cuda)" shell: bash - - # TODO: Include catboost in Python 3.14 CI when catboost supports it: - # https://github.com/catboost/catboost/issues/2943 - - name: Exclude catboost on Python 3.14 from pyproject.toml - if: ${{ inputs.python-version == '3.14' }} - run: | - sed -i '/ "catboost",/d' pyproject.toml - cat pyproject.toml - shell: bash diff --git a/test/gbdt/test_gbdt.py b/test/gbdt/test_gbdt.py index e4e2d5872..b216f04c5 100644 --- a/test/gbdt/test_gbdt.py +++ b/test/gbdt/test_gbdt.py @@ -1,5 +1,4 @@ import os.path as osp -import sys import tempfile import pytest @@ -13,21 +12,11 @@ from torch_frame.testing.text_embedder import HashTextEmbedder -@pytest.mark.parametrize( - 'gbdt_cls', - [ - # TODO: Run CatBoost test on Python 3.14 once supported - # https://github.com/catboost/catboost/issues/2943 - pytest.param( - CatBoost, - marks=pytest.mark.skipif( - sys.version_info >= (3, 14), - reason="Not supported on Python 3.14", - ), - ), - XGBoost, - LightGBM, - ]) +@pytest.mark.parametrize('gbdt_cls', [ + CatBoost, + XGBoost, + LightGBM, +]) @pytest.mark.parametrize('stypes', [ [stype.numerical], [stype.categorical],