Add extra xgboost

opcode81 · Aug 13, 2024 · 9c81fe3 · 9c81fe3
1 parent baf6720
commit 9c81fe3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/requirements_xgboost.txt 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
@@ -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] = {}

diff --git a/tox.ini b/tox.ini
@@ -34,6 +34,7 @@ deps =
     -rrequirements_lightgbm.txt
     -rrequirements_geoanalytics.txt
     -rrequirements_tensorflow.txt
+    -rrequirements_xgboost.txt
 whitelist_externals =
     sh
 allowlist_externals =