From e39fdc25e144a5168757a21333536f63c4dff066 Mon Sep 17 00:00:00 2001
From: Arne Tarara <arne@datafuse.de>
Date: Thu, 3 Oct 2024 11:24:04 +0200
Subject: [PATCH] Moving dev requirements to -dev.txt; Using xgboost-cpu on
 linux which is 10% of the size

---
 requirements-dev.txt | 2 ++
 requirements.txt     | 5 ++---
 2 files changed, 4 insertions(+), 3 deletions(-)
 create mode 100644 requirements-dev.txt

diff --git a/requirements-dev.txt b/requirements-dev.txt
new file mode 100644
index 0000000..3cd566b
--- /dev/null
+++ b/requirements-dev.txt
@@ -0,0 +1,2 @@
+optuna==4.0.0
+scikit-learn==1.5.2
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 1593d80..3d079ed 100644
--- 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
\ No newline at end of file