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] = {}