diff --git a/train/requirements.txt b/train/requirements.txt
index 02cba9f..392581c 100644
--- a/train/requirements.txt
+++ b/train/requirements.txt
@@ -7,7 +7,6 @@ scipy
 tensorflow-addons # for connected_components, depublished and only compatible with tensorflow < 2.16
 tensorflow < 2.16 # for tensorflow-addons, so only needed in training
 tf-keras < 2.16 # avoid keras 3 (also needs TF_USE_LEGACY_KERAS=1)
-tf-data < 2.16 # for tensorflow-addons, so only needed in training
 protobuf < 5 # for tensorflow-addons, so only needed in training
-eynollah-fork-tf2onnx == 1.17.0.post1
+eynollah-fork-tf2onnx == 1.17.0.post2
 ml_dtypes >= 0.5