diff --git a/pgml-extension/requirements.txt b/pgml-extension/requirements.txt
index 235c726ef..ba86e28f5 100644
--- a/pgml-extension/requirements.txt
+++ b/pgml-extension/requirements.txt
@@ -29,3 +29,4 @@ pynvml==11.5.0
 transformers-stream-generator==0.0.4
 optimum==1.13.2
 peft==0.6.2
+pyarrow==11.0.0