diff --git a/environment.yml b/environment.yml
index eeec142..73f1941 100644
--- a/environment.yml
+++ b/environment.yml
@@ -14,6 +14,7 @@ dependencies:
     - transformers>=4.40
     - torch>=2.2
     - accelerate>=0.27
+    - scikit-learn>=1.4
 
     # Optional: GLiClass adapter
     # - gliclass