From b9ba43b44499bd19609e78f7026c159c99daf13e Mon Sep 17 00:00:00 2001 From: kba Date: Thu, 9 Jul 2026 14:37:51 +0200 Subject: [PATCH] training: require tf2onnx and pin ml_dtypes >= 0.5 --- .gitignore | 2 ++ train/requirements.txt | 4 +++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/.gitignore b/.gitignore index 49835a7..e3356ea 100644 --- a/.gitignore +++ b/.gitignore @@ -12,3 +12,5 @@ output.html *.sw? TAGS uv.lock +/ignore +*.log diff --git a/train/requirements.txt b/train/requirements.txt index 1734b67..dbc4f76 100644 --- a/train/requirements.txt +++ b/train/requirements.txt @@ -6,6 +6,8 @@ imutils 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 # avoid keras 3 (also needs TF_USE_LEGACY_KERAS=1) +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 +tf2onnx +ml_dtypes >= 0.5