Merge pull request #1025 from qiboteam/remove_onnx

Remove onnx dep
qiboteam · Oct 29, 2024 · 74d2625 · 74d2625
2 parents cace06e + d433967
commit 74d2625
Show file tree

Hide file tree

Showing 14 changed files with 15 additions and 584 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,15 +28,10 @@ jinja2 = "^3.1.2"
 plotly = "^5.22.0"
 dash = "^2.6.0"
 skops = "^0.10.0"
-scikit-learn = "^1.2.1"
 matplotlib = { version = "^3.7.0", optional = true }
 seaborn = { version = "^0.12.2", optional = true }
 pydot = { version = "^1.4.2", optional = true }
-skl2onnx = { version = "^1.14.0", optional = true }
 pyyaml = "^6.0"
-onnxruntime = { version = "^1.14.1", optional = true }
-# TODO: drop this when https://github.com/onnx/onnx/issues/6267 is solved
-onnx = { version = "<1.16.2", optional = true }
 
 [tool.poetry.group.test]
 optional = true
@@ -69,7 +64,7 @@ devtools = "^0.10.0"
 
 [tool.poetry.extras]
 
-classify = ["matplotlib", "seaborn", "skl2onnx", "onnxruntime"]
+classify = ["matplotlib", "seaborn", "skl2onnx"]
 viz = ["pydot"]
 
 [build-system]

diff --git a/src/qibocal/fitting/classifier/ada_boost.py b/src/qibocal/fitting/classifier/ada_boost.py
diff --git a/src/qibocal/fitting/classifier/decision_tree.py b/src/qibocal/fitting/classifier/decision_tree.py
diff --git a/src/qibocal/fitting/classifier/gaussian_process.py b/src/qibocal/fitting/classifier/gaussian_process.py