From f9388df0ff77bd82aea46d8055951a64a1a2112a Mon Sep 17 00:00:00 2001 From: Dominik Jain Date: Tue, 13 Aug 2024 23:15:51 +0200 Subject: [PATCH] Add extra xgboost --- requirements_xgboost.txt | 3 +++ setup.py | 2 ++ 2 files changed, 5 insertions(+) create mode 100644 requirements_xgboost.txt diff --git a/requirements_xgboost.txt b/requirements_xgboost.txt new file mode 100644 index 00000000..b7039914 --- /dev/null +++ b/requirements_xgboost.txt @@ -0,0 +1,3 @@ +# Requirements for extra 'xgboost' +# These are the pinned lowest versions we test; setup.py relaxes them, changing them to lower bounds +xgboost==1.7.6 diff --git a/setup.py b/setup.py index a204fce6..50e5d999 100644 --- a/setup.py +++ b/setup.py @@ -19,6 +19,8 @@ "lightgbm", # extra "geoanalytics" "networkx", "Shapely", "geopandas", "utm", + # extra "xgboost" + "xgboost", ] # upper bound: map dependency name to lowest exluded version DEPS_VERSION_UPPER_BOUND_EXCLUSIVE: Dict[str, str] = {}