From 3c8a9802aadb08a5178ac2ec92a87edd9f9c6ea6 Mon Sep 17 00:00:00 2001 From: Aurelien FOUCRET Date: Wed, 7 Feb 2024 08:35:10 +0100 Subject: [PATCH 1/2] Adding support for xgboost 2.x --- docs/requirements-docs.txt | 2 +- requirements-dev.txt | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/requirements-docs.txt b/docs/requirements-docs.txt index 6b5dd7a5..e9d20727 100644 --- a/docs/requirements-docs.txt +++ b/docs/requirements-docs.txt @@ -3,7 +3,7 @@ pandas>=1.5 matplotlib>=3.6 nbval scikit-learn>=0.22.1 -xgboost>=1 +xgboost>=2 lightgbm sphinx==5.3.0 nbsphinx diff --git a/requirements-dev.txt b/requirements-dev.txt index 9a1308f0..f5b06231 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -11,7 +11,7 @@ tqdm<5 # Extras # scikit-learn>=1.3,<1.4 -xgboost>=0.90,<2 +xgboost>=0.90,<=2 lightgbm>=2,<4 # PyTorch doesn't support Python 3.11 yet (pytorch/pytorch#86566) diff --git a/setup.py b/setup.py index 4a3334e0..dd059b6c 100644 --- a/setup.py +++ b/setup.py @@ -54,7 +54,7 @@ long_description = "\n".join(lines[last_html_index:]) extras = { - "xgboost": ["xgboost>=0.90,<2"], + "xgboost": ["xgboost>=0.90,<=2"], "scikit-learn": ["scikit-learn>=1.3,<1.4"], "lightgbm": ["lightgbm>=2,<4"], "pytorch": [ From bef71bb07c750149fb104e387117fcafb6c57ef5 Mon Sep 17 00:00:00 2001 From: Aurelien FOUCRET Date: Wed, 7 Feb 2024 08:54:13 +0100 Subject: [PATCH 2/2] Fix xgboost version to be < 2.1 --- docs/requirements-docs.txt | 2 +- requirements-dev.txt | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/requirements-docs.txt b/docs/requirements-docs.txt index e9d20727..d0021f64 100644 --- a/docs/requirements-docs.txt +++ b/docs/requirements-docs.txt @@ -3,7 +3,7 @@ pandas>=1.5 matplotlib>=3.6 nbval scikit-learn>=0.22.1 -xgboost>=2 +xgboost<2.1 lightgbm sphinx==5.3.0 nbsphinx diff --git a/requirements-dev.txt b/requirements-dev.txt index f5b06231..d869748c 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -11,7 +11,7 @@ tqdm<5 # Extras # scikit-learn>=1.3,<1.4 -xgboost>=0.90,<=2 +xgboost>=0.90,<2.1 lightgbm>=2,<4 # PyTorch doesn't support Python 3.11 yet (pytorch/pytorch#86566) diff --git a/setup.py b/setup.py index dd059b6c..ce57e9f9 100644 --- a/setup.py +++ b/setup.py @@ -54,7 +54,7 @@ long_description = "\n".join(lines[last_html_index:]) extras = { - "xgboost": ["xgboost>=0.90,<=2"], + "xgboost": ["xgboost>=0.90, <2.1"], "scikit-learn": ["scikit-learn>=1.3,<1.4"], "lightgbm": ["lightgbm>=2,<4"], "pytorch": [