sdpython / pymlbenchmark

@@ -82,9 +82,9 @@
Loading
82 82
                            info.append(sub.replace(' ', ''))
83 83
                    obs['value'] = ", ".join(info)
84 84
                elif name == "onnx":
85 -
                    from mlprodict.tools.asv_options_helper import get_opset_number_from_onnx  # pylint: disable=C0415
85 +
                    from mlprodict import __max_supported_opset_  # pylint: disable=C0415
86 86
                    obs['value'] = "opset={}".format(
87 -
                        get_opset_number_from_onnx())
87 +
                        __max_supported_opset_)
88 88
                res.append(obs)
89 89
            else:
90 90
                res.append(dict(name=name, version='not-imported'))

@@ -11,8 +11,7 @@
Loading
11 11
from sklearn.ensemble._forest import BaseForest
12 12
from sklearn.tree._classes import BaseDecisionTree
13 13
from mlprodict.onnxrt import OnnxInference
14 -
from mlprodict.tools.asv_options_helper import (
15 -
    get_opset_number_from_onnx, get_ir_version_from_onnx)
14 +
from mlprodict import __max_supported_opset__, get_ir_version
16 15
from ..benchmark import BenchPerfTest
17 16
from ..benchmark.sklearn_helper import get_nb_skl_base_estimators
18 17
@@ -72,8 +71,8 @@
Loading
72 71
            with contextlib.redirect_stderr(self.logconvert):
73 72
                onx = to_onnx(self.skl, initial_types=initial_types,
74 73
                              options=onnx_options,
75 -
                              target_opset=get_opset_number_from_onnx())
76 -
                onx.ir_version = get_ir_version_from_onnx()
74 +
                              target_opset=__max_supported_opset__)
75 +
                onx.ir_version = get_ir_version(__max_supported_opset__)
77 76
78 77
        self._init(onx, runtimes)
79 78
@@ -120,7 +119,7 @@
Loading
120 119
        """
121 120
        self.onnx_info = {
122 121
            'onnx_nodes': len(self.ort_onnx.graph.node),  # pylint: disable=E1101
123 -
            'onnx_opset': get_opset_number_from_onnx(),
122 +
            'onnx_opset': __max_supported_opset__,
124 123
        }
125 124
        self.onnx_info.update(kwargs)
126 125
Files Coverage
src/pymlbenchmark 95.35%
Project Totals (20 files) 95.35%
Sunburst
The inner-most circle is the entire project, moving away from the center are folders then, finally, a single file. The size and color of each slice is representing the number of statements and the coverage, respectively.
Icicle
The top section represents the entire project. Proceeding with folders and finally individual files. The size and color of each slice is representing the number of statements and the coverage, respectively.
Grid
Each block represents a single file in the project. The size and color of each block is represented by the number of statements and the coverage, respectively.
Loading