diff --git a/requirements.txt b/requirements.txt
index 50b7e38..af8b4c3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,6 +11,7 @@ torch_geometric==2.6.1
 # Other dependencies
 joblib==1.3.2
 networkx==3.2.1
+numpy<2
 rdkit==2023.9.5
 # scikit_learn==1.4.1.post1
 scikit-learn>=1.2.2
@@ -26,4 +27,4 @@ optuna
 # furo
 
 # ogb
-# pytest
\ No newline at end of file
+# pytest