Skip to content

Commit

Permalink
[#161] Make the "xgboost is missing" error message more helpful
Browse files Browse the repository at this point in the history
  • Loading branch information
riley-harper committed Nov 18, 2024
1 parent 0277d7d commit 3ca3952
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion hlink/linking/core/classifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,9 @@ def choose_classifier(model_type, params, dep_var):
elif model_type == "xgboost":
if not _xgboost_available:
raise ModuleNotFoundError(
"model_type 'xgboost' requires the xgboost library"
"To use the experimental 'xgboost' model type, you need to install "
"the xgboost library and its dependencies. Try installing hlink with "
"the xgboost extra: 'pip install hlink[xgboost]'."
)
params_without_threshold = {
key: val
Expand Down

0 comments on commit 3ca3952

Please sign in to comment.