diff --git a/environment-dev.yml b/environment-dev.yml
index b6dfabc1..f20c7676 100644
--- a/environment-dev.yml
+++ b/environment-dev.yml
@@ -12,6 +12,7 @@ dependencies:
   - opencv==4.9.0
   - click==8.1.7
   - PyYAML==6.0.1
+  - scikit-learn==1.4.0
   - pip
 # dev dependencies
   - matplotlib==3.8.0
@@ -25,7 +26,6 @@ dependencies:
       - amazon-textract-textractor
       - backoff
       - PyMuPDF==1.23.26
-      - scikit-learn==1.4.1
       - -e .
 # dev pip dependencies
       - mlflow==2.11.1
\ No newline at end of file
diff --git a/environment-prod.yml b/environment-prod.yml
index 4c0cd5ae..66f28461 100644
--- a/environment-prod.yml
+++ b/environment-prod.yml
@@ -9,6 +9,7 @@ dependencies:
   - pathlib==1.0.1
   - python-dotenv==1.0.1
   - opencv==4.9.0
+  - scikit-learn==1.4.0
   - click==8.1.7
   - PyYAML==6.0.1
   - pip
@@ -16,5 +17,4 @@ dependencies:
       - amazon-textract-textractor
       - backoff
       - PyMuPDF==1.23.26
-      - scikit-learn==1.4.1
       - -e .