deps

xadupre · xadupre · commit c0fa5998350c · 2023-09-18T11:30:04.000+02:00
diff --git a/_doc/examples/plot_visualize_pipeline.py b/_doc/examples/plot_visualize_pipeline.py
@@ -64,7 +64,7 @@
 
 
 cmd = "dot -G=300 -Tpng {0} -o{0}.png".format(dot_file)
-run_cmd(cmd, wait=True, fLOG=print)
+run_cmd(cmd, wait=True)
 
 
 img = Image.open("graph.dot.png")
@@ -143,7 +143,7 @@
     f.write(dot)
 
 cmd = "dot -G=300 -Tpng {0} -o{0}.png".format(dot_file)
-run_cmd(cmd, wait=True, fLOG=print)
+run_cmd(cmd, wait=True)
 
 img = Image.open("graph2.dot.png")
 img
@@ -170,7 +170,7 @@
     f.write(dot)
 
 cmd = "dot -G=300 -Tpng {0} -o{0}.png".format(dot_file)
-run_cmd(cmd, wait=True, fLOG=print)
+run_cmd(cmd, wait=True)
 
 img = Image.open("graph3.dot.png")
 img
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,7 +20,7 @@ classifiers = [
 ]
 dependencies = ["numpy", "onnx>=1.14.0", "scipy"]
 description = "Extends the list of supported operators in onnx reference implementation and onnxruntime, or implements faster versions in C++."
-keywords = ["onnx", "onnxruntime", "CUDA", "openmp", "cmake", "cython", "pybind11"]
+keywords = ["onnx", "cmake", "cython", "scikit-learn", "machine-learning"]
 license = {file = "LICENSE.txt"}
 name = "mlinsights"
 readme = "README.rst"
@@ -55,11 +55,12 @@ dev = [
     "pytest",
     "pytest-cov",
     "ruff",
-    "scikit-learn",
+    "scikit-learn>=1.3.0",
     "skl2onnx>=1.14.1",
     "sphinx",
     "sphinx-gallery",
     "sphinx-issues",
+    "sphinx-runpython",
     "tqdm",
     "wheel",
 ]
@@ -70,6 +71,7 @@ requires = [
     "cmake",
     "numpy",
     "pybind11",
+    "scikit-learn>=1.3.0",
     "scipy",
     "setuptools",
     "wheel",
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -18,10 +18,14 @@ pytest-cov
 rstcheck[sphinx,toml]
 ruff
 seaborn
+skl2onnx>=1.14.1
 sphinx
 sphinx-gallery
 sphinx-issues
-git+https://github.com/sdpython/sphinx-runpython.git
+sphinx-runpython
 tomli
+torch
+torchvision
+torchaudio
 tqdm
 wheel
