diff --git a/plugins/flytekit-onnx-pytorch/dev-requirements.in b/plugins/flytekit-onnx-pytorch/dev-requirements.in index 808806fc24..638d377668 100644 --- a/plugins/flytekit-onnx-pytorch/dev-requirements.in +++ b/plugins/flytekit-onnx-pytorch/dev-requirements.in @@ -1,4 +1,4 @@ +numpy<2 onnxruntime pillow torchvision>=0.12.0 -numpy<2 \ No newline at end of file diff --git a/plugins/flytekit-onnx-scikitlearn/dev-requirements.in b/plugins/flytekit-onnx-scikitlearn/dev-requirements.in index c797056113..5d5bf2f3e8 100644 --- a/plugins/flytekit-onnx-scikitlearn/dev-requirements.in +++ b/plugins/flytekit-onnx-scikitlearn/dev-requirements.in @@ -1,2 +1,2 @@ -onnxruntime numpy<2 +onnxruntime