diff --git a/environment.yml b/environment.yml
index 5e363fc..6fe8151 100644
--- a/environment.yml
+++ b/environment.yml
@@ -24,3 +24,4 @@ dependencies:
   - pip:
     - ramp-workflow==0.6.0
     - osfclient
+    - optuna
diff --git a/requirements.txt b/requirements.txt
index b3f2d66..a21a3e4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,4 +8,5 @@ joblib
 osfclient
 ipywidgets
 ramp-workflow==0.6.0
+optuna
 imbalanced-learn