onnx 1.16.1__cp310-cp310-win_amd64.whl → 1.17.0__cp310-cp310-win_amd64.whl

onnx/reference/ops/aionnx_preview_training/op_momentum.py

@@ -1,6 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
+from __future__ import annotations
 
 from onnx.reference.ops.aionnx_preview_training._op_run_training import OpRunTraining
 
@@ -47,7 +48,7 @@ class Momentum(OpRunTraining):
         n = (len(data) - 2) // 3
         xs = []
         vs = []
-        for i in range(0, n):
+        for i in range(n):
             a, b = self._run1(  # type: ignore
                 *data[:2],
                 data[2 + i],
@@ -62,7 +63,18 @@ class Momentum(OpRunTraining):
             vs.append(b)
         return tuple(xs + vs)
 
-    def _run1(self, r, t, x, g, v, mode="standard", norm_coefficient=None, alpha=None, beta=None):  # type: ignore
+    def _run1(
+        self,
+        r,
+        t,
+        x,
+        g,
+        v,
+        mode="standard",
+        norm_coefficient=None,
+        alpha=None,
+        beta=None,
+    ):  # type: ignore
         if mode == "standard":
             x_new, v_new = _apply_momentum(r, t, x, g, v, norm_coefficient, alpha, beta)
         else:

onnx/reference/ops/aionnxml/__init__.py

@@ -1,3 +1,4 @@
 # SPDX-License-Identifier: Apache-2.0
+from __future__ import annotations
 
 from onnx.reference.ops.aionnxml._op_list import load_op

onnx/reference/ops/aionnxml/_common_classifier.py

@@ -1,4 +1,5 @@
 # SPDX-License-Identifier: Apache-2.0
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/aionnxml/_op_list.py

@@ -3,6 +3,7 @@
 # Operator ZipMap is not implemented. Its use should
 # be discouraged. It is just a different way to output
 # probabilites not consumed by any operator.
+from __future__ import annotations
 
 import textwrap
 from typing import Any, Dict
@@ -32,16 +33,16 @@ from onnx.reference.ops.aionnxml.op_tree_ensemble_classifier import (
 from onnx.reference.ops.aionnxml.op_tree_ensemble_regressor import TreeEnsembleRegressor
 
 
-def _build_registered_operators() -> Dict[str, Dict[Union[int, None], OpRunAiOnnxMl]]:
+def _build_registered_operators() -> dict[str, dict[int | None, OpRunAiOnnxMl]]:
     return build_registered_operators_any_domain(globals().copy())  # type: ignore[return-value]
 
 
 def load_op(
     domain: str,
     op_type: str,
-    version: Union[None, int],
+    version: None | int,
     custom: Any = None,
-    evaluator_cls: TOptional[type] = None,
+    evaluator_cls: type | None = None,  # noqa: ARG001
 ) -> Any:
     """Loads the implemented for a specified operator.
 
@@ -100,6 +101,4 @@ def load_op(
     return cl
 
 
-_registered_operators: TOptional[
-    Dict[str, Dict[Union[int, None], OpRunAiOnnxMl]]
-] = None
+_registered_operators: dict[str, dict[int | None, OpRunAiOnnxMl]] | None = None

onnx/reference/ops/aionnxml/_op_run_aionnxml.py

@@ -1,5 +1,5 @@
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from onnx.reference.op_run import OpRun
 

onnx/reference/ops/aionnxml/op_array_feature_extractor.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from onnx.reference.ops.aionnxml._op_run_aionnxml import OpRunAiOnnxMl
 

onnx/reference/ops/aionnxml/op_binarizer.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from onnx.reference.ops.aionnxml._op_run_aionnxml import OpRunAiOnnxMl
 

onnx/reference/ops/aionnxml/op_dict_vectorizer.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -50,7 +50,7 @@ class DictVectorizer(OpRunAiOnnxMl):
             keys = int64_vocabulary or string_vocabulary
             result = []
             for k in keys:
-                result.append(x.get(k, 0))
+                result.append(x.get(k, 0))  # noqa: PERF401
             return (np.array(result),)
 
         raise TypeError(f"x must be iterable not {type(x)}.")

onnx/reference/ops/aionnxml/op_feature_vectorizer.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/aionnxml/op_imputer.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -36,11 +36,11 @@ class Imputer(OpRunAiOnnxMl):
             )
         x = x.copy()
         if np.isnan(replace):
-            for i in range(0, x.shape[1]):
+            for i in range(x.shape[1]):
                 val = values[min(i, values.shape[0] - 1)]
                 x[np.isnan(x[:, i]), i] = val
         else:
-            for i in range(0, x.shape[1]):
+            for i in range(x.shape[1]):
                 val = values[min(i, values.shape[0] - 1)]
                 x[x[:, i] == replace, i] = val
 

onnx/reference/ops/aionnxml/op_label_encoder.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/aionnxml/op_linear_classifier.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -30,7 +30,7 @@ class LinearClassifier(OpRunAiOnnxMl):
         classlabels_strings=None,
         coefficients=None,
         intercepts=None,
-        multi_class=None,
+        multi_class=None,  # noqa: ARG002
         post_transform=None,
     ):
         # multi_class is unused

onnx/reference/ops/aionnxml/op_linear_regressor.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/aionnxml/op_normalizer.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/aionnxml/op_one_hot_encoder.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/aionnxml/op_scaler.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from onnx.reference.ops.aionnxml._op_run_aionnxml import OpRunAiOnnxMl
 

onnx/reference/ops/aionnxml/op_svm_classifier.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -24,7 +24,7 @@ def multiclass_probability(k, R):
     P = np.empty((k,), dtype=R.dtype)
     eps = 0.005 / k
 
-    for t in range(0, k):
+    for t in range(k):
         P[t] = 1.0 / k
         Q[t, t] = (R[:t, t] ** 2).sum()
         Q[t, :t] = Q[:t, t]
@@ -38,7 +38,7 @@ def multiclass_probability(k, R):
         pQp = (P * Qp).sum()
 
         max_error = 0
-        for t in range(0, k):
+        for t in range(k):
             error = np.abs(Qp[t] - pQp)
             if error > max_error:
                 max_error = error
@@ -135,9 +135,9 @@ class SVMClassifier(OpRunAiOnnxMl):
     ):
         evals = 0
 
-        kernels_list = []
-        for j in range(vector_count_):
-            kernels_list.append(self._svm.kernel_dot(X, sv[j], kernel_type_))
+        kernels_list = [
+            self._svm.kernel_dot(X, sv[j], kernel_type_) for j in range(vector_count_)
+        ]
         kernels = np.array(kernels_list)
 
         votes = np.zeros((class_count_,), dtype=X.dtype)
@@ -214,7 +214,10 @@ class SVMClassifier(OpRunAiOnnxMl):
             label = max_class
 
         new_scores = write_scores(
-            scores.size, scores, self._svm.atts.post_transform, write_additional_scores  # type: ignore
+            scores.size,
+            scores,
+            self._svm.atts.post_transform,
+            write_additional_scores,  # type: ignore
         )
         return label, new_scores
 

onnx/reference/ops/aionnxml/op_svm_helper.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from typing import Any
 
@@ -23,7 +23,7 @@ class SVMAttributes:
     def __str__(self) -> str:
         rows = ["Attributes"]
         for name in self._names:
-            rows.append(f"  {name}={getattr(self, name)}")
+            rows.append(f"  {name}={getattr(self, name)}")  # noqa: PERF401
         return "\n".join(rows)
 
 

onnx/reference/ops/aionnxml/op_svm_regressor.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from onnx.reference.ops.aionnxml._op_run_aionnxml import OpRunAiOnnxMl
 from onnx.reference.ops.aionnxml.op_svm_helper import SVMCommon

onnx/reference/ops/aionnxml/op_tree_ensemble.py

@@ -39,7 +39,7 @@ class Leaf:
         self.target_id = target_id
 
     # Produce the weight and target index
-    def predict(self, x: np.ndarray) -> np.ndarray:
+    def predict(self, x: np.ndarray) -> np.ndarray:  # noqa: ARG002
         return np.array([self.weight, self.target_id])
 
     def _print(self, prefix: list, indent: int = 0) -> None:
@@ -132,9 +132,9 @@ class TreeEnsemble(OpRunAiOnnxMl):
         leaf_targetids,
         leaf_weights,
         tree_roots,
-        post_transform=PostTransform.NONE,
+        post_transform=PostTransform.NONE,  # noqa: ARG002
         aggregate_function=AggregationFunction.SUM,
-        nodes_hitrates=None,
+        nodes_hitrates=None,  # noqa: ARG002
         nodes_missing_value_tracks_true=None,
         membership_values=None,
         n_targets=None,

onnx/reference/ops/aionnxml/op_tree_ensemble_classifier.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/aionnxml/op_tree_ensemble_helper.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -100,6 +100,6 @@ class TreeEnsemble:
         for row in X:
             outs = []
             for tree_id in self.tree_ids:
-                outs.append(self.leaf_index_tree(row, tree_id))
+                outs.append(self.leaf_index_tree(row, tree_id))  # noqa: PERF401
             outputs.append(outs)
         return np.array(outputs)

onnx/reference/ops/aionnxml/op_tree_ensemble_regressor.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -80,14 +80,16 @@ class TreeEnsembleRegressor(OpRunAiOnnxMl):
                 for its in t_index:
                     for it in its:
                         res[i, target_ids[it]] = min(
-                            res[i, target_ids[it]], tr.atts.target_weights[it]  # type: ignore
+                            res[i, target_ids[it]],
+                            tr.atts.target_weights[it],  # type: ignore
                         )
             elif aggregate_function == "MAX":
                 res[i, :] = np.finfo(res.dtype).min
                 for its in t_index:
                     for it in its:
                         res[i, target_ids[it]] = max(
-                            res[i, target_ids[it]], tr.atts.target_weights[it]  # type: ignore
+                            res[i, target_ids[it]],
+                            tr.atts.target_weights[it],  # type: ignore
                         )
             else:
                 raise NotImplementedError(

onnx/reference/ops/experimental/__init__.py

@@ -1,5 +1,6 @@
 # Copyright (c) ONNX Project Contributors
 #
 # SPDX-License-Identifier: Apache-2.0
+from __future__ import annotations
 
 from onnx.reference.ops.experimental._op_list import load_op

onnx/reference/ops/experimental/_op_list.py

@@ -1,12 +1,10 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import textwrap
-from typing import Any, Dict
-from typing import Optional as TOptional
-from typing import Union
+from typing import Any
 
 from onnx.reference.op_run import OpFunction
 from onnx.reference.ops._helpers import build_registered_operators_any_domain
@@ -14,18 +12,16 @@ from onnx.reference.ops.experimental._op_run_experimental import OpRunExperiment
 from onnx.reference.ops.experimental.op_im2col import Im2Col  # noqa: F401
 
 
-def _build_registered_operators() -> (
-    Dict[str, Dict[Union[int, None], OpRunExperimental]]
-):
+def _build_registered_operators() -> dict[str, dict[int | None, OpRunExperimental]]:
     return build_registered_operators_any_domain(globals().copy())  # type: ignore[return-value]
 
 
 def load_op(
     domain: str,
     op_type: str,
-    version: Union[None, int],
+    version: None | int,
     custom: Any = None,
-    evaluator_cls: TOptional[type] = None,
+    evaluator_cls: type | None = None,  # noqa: ARG001
 ) -> Any:
     """Loads the implemented for a specified operator.
 
@@ -84,6 +80,4 @@ def load_op(
     return cl
 
 
-_registered_operators: TOptional[
-    Dict[str, Dict[Union[int, None], OpRunExperimental]]
-] = None
+_registered_operators: dict[str, dict[int | None, OpRunExperimental]] | None = None

onnx/reference/ops/experimental/_op_run_experimental.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 #
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from onnx.reference.op_run import OpRun
 

onnx/reference/ops/experimental/op_im2col.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 #
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from onnx.reference.ops.experimental._op_run_experimental import OpRunExperimental
 from onnx.reference.ops_optimized.op_conv_optimized import im2col_fast

onnx/reference/ops/op_abs.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_acos.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_acosh.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_add.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_affine_grid.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_and.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_argmax.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_argmin.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_asin.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_asinh.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_atan.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_atanh.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 

onnx/reference/ops/op_attribute_has_value.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -11,20 +11,20 @@ from onnx.reference.op_run import OpRun
 class AttributeHasValue(OpRun):
     def _run(  # type: ignore
         self,
-        value_float=None,
-        value_floats=None,
-        value_graph=None,
-        value_graphs=None,
-        value_int=None,
-        value_ints=None,
-        value_sparse_tensor=None,
-        value_sparse_tensors=None,
-        value_string=None,
-        value_strings=None,
-        value_tensor=None,
-        value_tensors=None,
-        value_type_proto=None,
-        value_type_protos=None,
+        value_float=None,  # noqa: ARG002
+        value_floats=None,  # noqa: ARG002
+        value_graph=None,  # noqa: ARG002
+        value_graphs=None,  # noqa: ARG002
+        value_int=None,  # noqa: ARG002
+        value_ints=None,  # noqa: ARG002
+        value_sparse_tensor=None,  # noqa: ARG002
+        value_sparse_tensors=None,  # noqa: ARG002
+        value_string=None,  # noqa: ARG002
+        value_strings=None,  # noqa: ARG002
+        value_tensor=None,  # noqa: ARG002
+        value_tensors=None,  # noqa: ARG002
+        value_type_proto=None,  # noqa: ARG002
+        value_type_protos=None,  # noqa: ARG002
     ):
         # TODO: support overridden attributes.
         for att in self.onnx_node.attribute:

onnx/reference/ops/op_average_pool.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from onnx.reference.ops.op_pool_common import CommonPool