add extras for better pip installation

tryolabs · Mar 5, 2024 · a00981f · a00981f
1 parent 5f273be
commit a00981f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 30 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,10 +21,13 @@ pandas = "^2.2.1"
 scikit-learn = "^1.4.1.post1"
 explainerdashboard = "^0.4.5"
 optuna = "^3.5.0"
-xgboost = "^2.0.3"
 pyarrow = "^15.0.0"
 PyYAML = "^6.0.1"
 
+[tool.poetry.extras]
+xgboost = ["xgboost>=2.0.0,<2.1.0"]
+all_models = ["xgboost>=2.0.0,<2.1.0"]
+
 [tool.poetry.group.dev.dependencies]
 flake8 = "^7.0.0"
 black = "^23.12.1"