From 7d0e5ec5def33b204b7ac51aadd4e235c7f0b69c Mon Sep 17 00:00:00 2001
From: Emilie Delattre <emilie.delattre@epfl.ch>
Date: Wed, 28 Sep 2022 16:25:32 +0200
Subject: [PATCH] Justify xgboost version bounds

---
 setup.cfg | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/setup.cfg b/setup.cfg
index 6a0083a..8d38fa5 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -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]