diff --git a/requirements/optional.txt b/requirements/optional.txt
index a5b95cc0..8ef39eb3 100644
--- a/requirements/optional.txt
+++ b/requirements/optional.txt
@@ -4,5 +4,5 @@ hyperopt==0.2.5
 mlflow==1.21.0
 nevergrad==0.4.3.post7
 optuna==2.10.0
+scikit-learn
 scikit-optimize==0.9.0
-sklearn
diff --git a/requirements/runtime.txt b/requirements/runtime.txt
index 7cf3a24b..d9ef98e1 100644
--- a/requirements/runtime.txt
+++ b/requirements/runtime.txt
@@ -1,5 +1,6 @@
 mmcv-full>=1.4.7
 pandas
 protobuf<=3.20
+pyarrow
 ray[default]==2.1.0
 tabulate