Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions setup.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,10 @@ install_requires =
torch
tqdm
umap-learn
; With the version xgboost==1.2.1 the loading of the DVC checkpoint fails
; torch.load("models/pyramidal_cells/L2/pc_tmd_L2_stratifKFold_xgb_oversampled_results.chk")
; With the version xgboost==1.5.0 there's a different error while running
; dvc repro -fs explain-models@pc_tmd_L2_stratifKFold_xgb_oversampled
xgboost>1.2.1,<1.5.0

[options.package_data]
Expand Down