diff --git a/train/requirements.txt b/train/requirements.txt
index 4df9c2f..63f3813 100644
--- a/train/requirements.txt
+++ b/train/requirements.txt
@@ -1,6 +1,6 @@
-# tensorflow == 2.12.1 # TODO why not tensorflow < 2.13 as in eynollah/requirements.txt
 sacred
 seaborn
+numpy <1.24.0
 tqdm
 imutils
 scipy