Add extra xgboost

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

Hide file tree

Showing 2 changed files with 5 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] = {}