Fix xgboost version conflict

antmicro · Dec 24, 2022 · 73aa645 · 73aa645
1 parent 7322178
commit 73aa645
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/environments/Dockerfile b/environments/Dockerfile
@@ -53,7 +53,7 @@ RUN $PIPINST \
     tensorflow-probability==0.18.0 \
     tflite \
     torchvision==0.13.1 \
-    xgboost
+    xgboost>=1.1.0,<1.6.0
 
 RUN $PIPINST git+https://github.com/onnx/[email protected]
 RUN $PIPINST git+https://github.com/onnx/[email protected]

diff --git a/environments/cfu-playground-environment.def b/environments/cfu-playground-environment.def
@@ -80,7 +80,7 @@ From: debian:bullseye
         tflite \
         tornado \
         tqdm \
-        xgboost
+        xgboost>=1.1.0,<1.6.0
 
     cd /
     wget https://static.dev.sifive.com/dev-tools/freedom-tools/v2020.08/riscv64-unknown-elf-gcc-10.1.0-2020.08.2-x86_64-linux-ubuntu14.tar.gz

diff --git a/environments/development-environment-gpu.def b/environments/development-environment-gpu.def
@@ -67,7 +67,7 @@ From: nvidia/cuda:11.4.2-cudnn8-devel-ubuntu20.04
         torchvision==0.13.1 \
         tornado \
         tqdm \
-        xgboost
+        xgboost>=1.1.0,<1.6.0
 
     $PIPINST git+https://github.com/onnx/[email protected]
     $PIPINST git+https://github.com/onnx/[email protected]

diff --git a/environments/development-environment.def b/environments/development-environment.def
@@ -58,7 +58,7 @@ From: archlinux:latest
         tensorflow-probability==0.18.0 \
         tflite \
         torchvision==0.13.1 \
-        xgboost
+        xgboost>=1.1.0,<1.6.0
 
     $PIPINST git+https://github.com/onnx/[email protected]
     $PIPINST git+https://github.com/onnx/[email protected]