diff --git a/environment.yml b/environment.yml
index 549f4ea..3061cf8 100644
--- a/environment.yml
+++ b/environment.yml
@@ -7,4 +7,5 @@ dependencies:
   - scikit-learn=1.4.1.post1
   - joblib=1.3.2
   - matplotlib=3.8.3
-  - flaml=2.1.2
\ No newline at end of file
+  - flaml=2.1.2
+  - pyarrow=16.1.0
\ No newline at end of file