training: require tf2onnx and pin ml_dtypes >= 0.5

This commit is contained in:
kba 2026-07-09 14:37:51 +02:00
parent c9c14ed83d
commit b9ba43b444
2 changed files with 5 additions and 1 deletions

2
.gitignore vendored
View file

@ -12,3 +12,5 @@ output.html
*.sw?
TAGS
uv.lock
/ignore
*.log

View file

@ -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