adjust poetry extras

tryolabs · Mar 6, 2024 · 871bbec · 871bbec
1 parent e4ddec5
commit 871bbec
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 8 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -23,10 +23,11 @@ explainerdashboard = "^0.4.5"
 optuna = "^3.5.0"
 pyarrow = "^15.0.0"
 PyYAML = "^6.0.1"
+xgboost = { version = "^2.0.3", optional = true }
 
 [tool.poetry.extras]
-xgboost = ["xgboost>=2.0.0,<2.1.0"]
-all_models = ["xgboost>=2.0.0,<2.1.0"]
+xgboost = ["xgboost"]
+all_models = ["xgboost"]
 
 [tool.poetry.group.dev.dependencies]
 flake8 = "^7.0.0"