Moving dev requirements to -dev.txt; Using xgboost-cpu on linux which…

… is 10% of the size
green-coding-solutions · Oct 3, 2024 · e39fdc2 · e39fdc2
1 parent 48e9e78
commit e39fdc2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -0,0 +1,2 @@
+optuna==4.0.0
+scikit-learn==1.5.2
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,6 @@
 numpy==2.1.1
-optuna==4.0.0
 pandas==2.2.3
-xgboost==2.1.1
+xgboost-cpu==2.1.1; platform_machine == "x86_64"
+xgboost==2.1.1; platform_machine != "x86_64"
 psutil==6.0.0
 pyarrow==17.0.0
-scikit-learn==1.5.2