onnx 1.16.1__cp39-cp39-win_amd64.whl → 1.17.0__cp39-cp39-win_amd64.whl

onnx/reference/ops/op_einsum.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_elu.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_equal.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_erf.py

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

onnx/reference/ops/op_exp.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_expand.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_eyelike.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,7 +11,7 @@ from onnx.reference.op_run import OpRun
 
 
 class EyeLike(OpRun):
-    def _run(self, data, *args, dtype=None, k=None):
+    def _run(self, data, *args, dtype=None, k=None):  # noqa: ARG002
         if dtype is None:
             if data is None:
                 _dtype = np.float32

onnx/reference/ops/op_flatten.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_floor.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_gather.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_gather_elements.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -28,10 +28,10 @@ def gather_numpy(self: np.ndarray, dim: int, index: np.ndarray) -> np.ndarray:
 
     try:
         gathered = np.choose(index_swaped, data_swaped, mode="wrap")
-    except ValueError as e:
+    except ValueError:
         if len(index_swaped.shape) == 2 and len(data_swaped.shape) == 2:
             return gather_numpy_2(self, index)
-        raise e  # pragma: no cover
+        raise  # pragma: no cover
 
     return np.swapaxes(gathered, 0, dim)
 

onnx/reference/ops/op_gathernd.py

@@ -1,9 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
-
-from typing import Tuple
+from __future__ import annotations
 
 import numpy as np
 
@@ -12,7 +10,7 @@ from onnx.reference.op_run import OpRun
 
 def _gather_nd_impl(
     data: np.ndarray, indices: np.ndarray, batch_dims: int
-) -> Tuple[np.ndarray]:
+) -> tuple[np.ndarray]:
     # Note the data rank - will be reused multiple times later
     data_rank = len(data.shape)
 

onnx/reference/ops/op_gemm.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -37,7 +37,17 @@ def _gemm11(a, b, c, alpha, beta):  # type: ignore
 
 
 class Gemm_6(OpRun):
-    def _run(self, a, b, c=None, alpha=None, beta=None, transA=None, transB=None, broadcast=None):  # type: ignore
+    def _run(
+        self,
+        a,
+        b,
+        c=None,
+        alpha=None,
+        beta=None,
+        transA=None,
+        transB=None,
+        broadcast=None,
+    ):  # type: ignore
         if broadcast == 0:
             if transA:
                 _meth = _gemm11 if transB else _gemm10

onnx/reference/ops/op_global_average_pool.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_global_max_pool.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_greater.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_greater_or_equal.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_grid_sample.py

@@ -1,10 +1,9 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numbers
-from typing import List
 
 import numpy as np
 
@@ -205,7 +204,7 @@ class GridSample(OpRun):
             return hi
         return val
 
-    def _pixel_at_ndarray(self, ndarray, x: List, border, padding_mode):  # type: ignore
+    def _pixel_at_ndarray(self, ndarray, x: list, border, padding_mode):  # type: ignore
         # boarder: [x_1_min, x_2_min, ..., x_1_max, x_2_max, ...]
         num_dims = ndarray.ndim
         assert num_dims == len(x) == int(len(border) / 2)

onnx/reference/ops/op_gru.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -77,14 +77,14 @@ class CommonGRU(OpRun):
         B=None,
         sequence_lens=None,
         initial_h=None,
-        activation_alpha=None,
-        activation_beta=None,
-        activations=None,
-        clip=None,
-        direction=None,
+        activation_alpha=None,  # noqa: ARG002
+        activation_beta=None,  # noqa: ARG002
+        activations=None,  # noqa: ARG002
+        clip=None,  # noqa: ARG002
+        direction=None,  # noqa: ARG002
         hidden_size=None,
         layout=None,
-        linear_before_reset=None,
+        linear_before_reset=None,  # noqa: ARG002
     ):
         # TODO: support overridden attributes.
         num_directions = W.shape[0]

onnx/reference/ops/op_hamming_window.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_hann_window.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_hard_sigmoid.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_hardmax.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -14,6 +14,9 @@ class Hardmax(OpRunUnaryNum):
         x_argmax = np.argmax(x, axis=axis)  # type: ignore
         y = np.zeros_like(x)
         np.put_along_axis(
-            y, np.expand_dims(x_argmax, axis=axis), 1, axis=axis  # type: ignore
+            y,
+            np.expand_dims(x_argmax, axis=axis),
+            1,
+            axis=axis,  # type: ignore
         )
         return (y,)

onnx/reference/ops/op_identity.py

@@ -1,12 +1,12 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
+from __future__ import annotations
 
+from onnx.reference.op_run import OpRun
 
-from onnx.reference.ops._op import OpRunUnaryNum
 
-
-class Identity(OpRunUnaryNum):
+class Identity(OpRun):
     def _run(self, a):  # type: ignore
         if a is None:
             return (None,)

onnx/reference/ops/op_if.py

@@ -31,8 +31,8 @@ class If(OpRun):
         self,
         cond: np.ndarray | np.bool_,
         context=None,
-        else_branch=None,
-        then_branch=None,
+        else_branch=None,  # noqa: ARG002
+        then_branch=None,  # noqa: ARG002
         attributes=None,
     ):
         if cond.size != 1:

onnx/reference/ops/op_instance_normalization.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_isinf.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_isnan.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_layer_normalization.py

@@ -1,9 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
-
-from typing import Tuple
+from __future__ import annotations
 
 import numpy as np
 
@@ -16,7 +14,7 @@ def _layer_normalization(
     B: np.ndarray,
     axis: int = -1,
     epsilon: float = 1e-5,
-) -> Tuple[np.ndarray, np.ndarray, np.ndarray]:
+) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
     X_shape = X.shape
     X_rank = len(X_shape)
     if axis < 0:

onnx/reference/ops/op_leaky_relu.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_less.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_less_or_equal.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_log.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_log_softmax.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_loop.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -55,7 +55,9 @@ class Loop(OpRun):
         while cond and (M is None or it < M):
             self._log("  -- loop> {%r}", context)
             if len(body.input_names) > 0 and body.input_names[0] is not None:
-                inputs[body.input_names[0]] = np.array(it, dtype=None if M is None else M.dtype)  # type: ignore
+                inputs[body.input_names[0]] = np.array(
+                    it, dtype=None if M is None else M.dtype
+                )  # type: ignore
             if len(body.input_names) > 1 and body.input_names[1] is not None:
                 inputs[body.input_names[1]] = cond
             outputs = self._run_body(inputs, attributes=attributes)  # type: ignore

onnx/reference/ops/op_lp_normalization.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_lp_pool.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -38,4 +38,6 @@ class LpPool(CommonPool):
         )
 
         kernel_element_count = np.prod(kernel_shape)
-        return (np.power(kernel_element_count * power_average[0], 1.0 / p),)
+        return (
+            np.power(kernel_element_count * power_average[0], 1.0 / p).astype(x.dtype),
+        )

onnx/reference/ops/op_lrn.py

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

onnx/reference/ops/op_lstm.py

@@ -1,9 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
-
-from typing import Tuple
+from __future__ import annotations
 
 import numpy as np
 
@@ -35,7 +33,7 @@ class CommonLSTM(OpRun):
         C_0: np.ndarray,
         P: np.ndarray,
         num_directions: int,
-    ) -> Tuple[np.ndarray, np.ndarray]:
+    ) -> tuple[np.ndarray, np.ndarray]:
         seq_length = X.shape[0]
         hidden_size = H_0.shape[-1]
         batch_size = X.shape[1]
@@ -85,14 +83,14 @@ class CommonLSTM(OpRun):
         initial_h=None,
         initial_c=None,
         P=None,
-        activation_alpha=None,
-        activation_beta=None,
-        activations=None,
-        clip=None,
-        direction=None,
+        activation_alpha=None,  # noqa: ARG002
+        activation_beta=None,  # noqa: ARG002
+        activations=None,  # noqa: ARG002
+        clip=None,  # noqa: ARG002
+        direction=None,  # noqa: ARG002
         hidden_size=None,
-        input_forget=None,
-        layout=None,
+        input_forget=None,  # noqa: ARG002
+        layout=None,  # noqa: ARG002
     ):
         # TODO: support overridden attributes.
         n_gates = 4

onnx/reference/ops/op_matmul.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_matmul_integer.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_max.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_max_pool.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -182,12 +182,12 @@ class MaxPool(CommonPool):
     def _max_pool_1d(  # type: ignore
         self,
         x,
-        auto_pad,
-        ceil_mode,
+        auto_pad,  # noqa: ARG002
+        ceil_mode,  # noqa: ARG002
         dilations,
         kernel_shape,
         new_pads,
-        storage_order,
+        storage_order,  # noqa: ARG002
         strides,
         output_spatial_shape,
     ):
@@ -238,8 +238,8 @@ class MaxPool(CommonPool):
     def _max_pool_2d(  # type: ignore
         self,
         x,
-        auto_pad,
-        ceil_mode,
+        auto_pad,  # noqa: ARG002
+        ceil_mode,  # noqa: ARG002
         dilations,
         kernel_shape,
         new_pads,
@@ -315,8 +315,8 @@ class MaxPool(CommonPool):
     def _max_pool_3d(  # type: ignore
         self,
         x,
-        auto_pad,
-        ceil_mode,
+        auto_pad,  # noqa: ARG002
+        ceil_mode,  # noqa: ARG002
         dilations,
         kernel_shape,
         new_pads,

onnx/reference/ops/op_max_unpool.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -9,7 +9,9 @@ from onnx.reference.op_run import OpRun
 
 
 class MaxUnpool(OpRun):
-    def _run(self, X, indices, output_shape=None, kernel_shape=None, pads=None, strides=None):  # type: ignore
+    def _run(
+        self, X, indices, output_shape=None, kernel_shape=None, pads=None, strides=None
+    ):  # type: ignore
         pooling_dims = len(X.shape) - 2
         if pooling_dims > 3:
             raise NotImplementedError(
@@ -28,7 +30,7 @@ class MaxUnpool(OpRun):
         inferred_shape[0] = X.shape[0]
         inferred_shape[1] = X.shape[1]
 
-        for dim in range(0, len(kernel_shape)):
+        for dim in range(len(kernel_shape)):
             inferred_shape[dim + 2] = (
                 (X.shape[dim + 2] - 1) * strides[dim]
                 - (pads[dim] + pads[len(kernel_shape) + dim])

onnx/reference/ops/op_mean.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