Improve `scikit-learn` support for exports (#16638)

2 months ago · bf51faeac5
parent 5c76601d21
commit bf51faeac5
1 changed files with 1 additions and 0 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@ -98,6 +98,7 @@ dev = [
 export = [
    "onnx>=1.12.0", # ONNX export
    "coremltools>=7.0; platform_system != 'Windows' and python_version <= '3.11'", # CoreML supported on macOS and Linux
+    "scikit-learn>=1.3.2; platform_system != 'Windows' and python_version <= '3.11'", # CoreML k-means quantization
    "openvino>=2024.0.0", # OpenVINO export
    "tensorflow>=2.0.0", # TF bug https://github.com/ultralytics/ultralytics/issues/5161
    "tensorflowjs>=3.9.0", # TF.js export, automatically installs tensorflow