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

onnx/reference/ops/op_sequence_length.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_sequence_map.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/op_shape.py

@@ -1,9 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
-
-from typing import Optional, Tuple
+from __future__ import annotations
 
 import numpy as np
 
@@ -17,9 +15,7 @@ class Shape_1(OpRun):
 
 class Shape_15(Shape_1):
     @staticmethod
-    def _interval(
-        n: int, start: Optional[int], end: Optional[int]
-    ) -> Optional[Tuple[int, int]]:
+    def _interval(n: int, start: int | None, end: int | None) -> tuple[int, int] | None:
         if start == 0:
             if end is None or np.isnan(end):
                 return None

onnx/reference/ops/op_shrink.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_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_sign.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_sin.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_sinh.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_size.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_slice.py

@@ -1,9 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
-
-from typing import Optional
+from __future__ import annotations
 
 import numpy as np
 
@@ -14,8 +12,8 @@ def _slice(
     data: np.ndarray,
     starts: np.ndarray,
     ends: np.ndarray,
-    axes: Optional[np.ndarray] = None,
-    steps: Optional[np.ndarray] = None,
+    axes: np.ndarray | None = None,
+    steps: np.ndarray | None = None,
 ) -> np.ndarray:
     if isinstance(starts, list):
         starts = np.array(starts)

onnx/reference/ops/op_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_softmax_cross_entropy_loss.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_softplus.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_softsign.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_space_to_depth.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_split.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/op_split_to_sequence.py

@@ -1,9 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
-
-from typing import List, Optional, Tuple
+from __future__ import annotations
 
 import numpy as np
 
@@ -12,8 +10,8 @@ from onnx.reference.op_run import OpRun
 
 class SplitToSequence(OpRun):
     def common_run(
-        self, mat: np.ndarray, split: Optional[np.ndarray], axis: int
-    ) -> List[np.ndarray]:
+        self, mat: np.ndarray, split: np.ndarray | None, axis: int
+    ) -> list[np.ndarray]:
         if split is None:
             split_length = [1 for _ in range(mat.shape[axis])]
         elif len(split.shape) == 0:
@@ -40,10 +38,10 @@ class SplitToSequence(OpRun):
     def _run(
         self,
         mat: np.ndarray,
-        split: Optional[np.ndarray] = None,
+        split: np.ndarray | None = None,
         axis: int = 0,
         keepdims: int = 1,
-    ) -> Tuple[np.ndarray]:
+    ) -> tuple[np.ndarray]:
         res = self.common_run(mat, split, axis=axis)
         if split is None and not keepdims:
             for i in range(len(res)):

onnx/reference/ops/op_sqrt.py

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

onnx/reference/ops/op_squeeze.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_stft.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -91,7 +91,8 @@ def _istft(x, fft_length: int, hop_length, window, onesided=False):  # type: ign
         begin = fs
         end = fs + 1
         frame_x = np.squeeze(
-            _slice(x, np.array([begin]), np.array([end]), axisf), axis=axisf[0]  # type: ignore
+            _slice(x, np.array([begin]), np.array([end]), axisf),
+            axis=axisf[0],  # type: ignore
         )
 
         # ifft

onnx/reference/ops/op_string_concat.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_string_normalizer.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import locale as pylocale
 import unicodedata
@@ -41,7 +41,7 @@ class StringNormalizer(OpRun):
                 stops = set(stopwords)
         res = np.empty(x.shape, dtype=x.dtype)
         if len(x.shape) == 2:
-            for i in range(0, x.shape[1]):
+            for i in range(x.shape[1]):
                 self._run_column(
                     x[:, i],
                     res[:, i],
@@ -93,7 +93,7 @@ class StringNormalizer(OpRun):
                 )
         cout[:] = cin[:]
 
-        for i in range(0, cin.shape[0]):
+        for i in range(cin.shape[0]):
             if isinstance(cout[i], float):
                 # nan
                 cout[i] = ""
@@ -101,14 +101,14 @@ class StringNormalizer(OpRun):
                 cout[i] = StringNormalizer.strip_accents_unicode(cout[i])
 
         if is_case_sensitive and len(stops) > 0:
-            for i in range(0, cin.shape[0]):
+            for i in range(cin.shape[0]):
                 cout[i] = StringNormalizer._remove_stopwords(cout[i], raw_stops)
 
         if case_change_action == "LOWER":
-            for i in range(0, cin.shape[0]):
+            for i in range(cin.shape[0]):
                 cout[i] = cout[i].lower()
         elif case_change_action == "UPPER":
-            for i in range(0, cin.shape[0]):
+            for i in range(cin.shape[0]):
                 cout[i] = cout[i].upper()
         elif case_change_action != "NONE":
             raise RuntimeError(
@@ -116,7 +116,7 @@ class StringNormalizer(OpRun):
             )
 
         if not is_case_sensitive and len(stops) > 0:
-            for i in range(0, cin.shape[0]):
+            for i in range(cin.shape[0]):
                 cout[i] = StringNormalizer._remove_stopwords(cout[i], stops)
 
         return cout
@@ -143,7 +143,7 @@ class StringNormalizer(OpRun):
             # If `s` is ASCII-compatible, then it does not contain any accented
             # characters and we can avoid an expensive list comprehension
             s.encode("ASCII", errors="strict")
-            return s
+            return s  # noqa: TRY300
         except UnicodeEncodeError:
             normalized = unicodedata.normalize("NFKD", s)
             s = "".join([c for c in normalized if not unicodedata.combining(c)])

onnx/reference/ops/op_string_split.py

@@ -1,9 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
-
-from typing import Union
+from __future__ import annotations
 
 import numpy as np
 
@@ -13,7 +11,7 @@ _acceptable_str_dtypes = ("U", "O")
 
 
 def pad_empty_string(
-    split_lists: Union[list, np.ndarray], padding_requirement: Union[list, int]
+    split_lists: list | np.ndarray, padding_requirement: list | int
 ) -> list:
     if isinstance(split_lists, list):
         return split_lists + ["" for _ in range(padding_requirement)]

onnx/reference/ops/op_sub.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_sum.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/op_tan.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_tanh.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_tfidf_vectorizer.py

@@ -1,10 +1,9 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 from enum import IntEnum
-from typing import List
 
 import numpy as np
 
@@ -183,7 +182,7 @@ class TfIdfVectorizer(OpRun):
             ngram_size += 1
 
     def increment_count(
-        self, ngram_id: int, row_num: int, frequencies: List[int]
+        self, ngram_id: int, row_num: int, frequencies: list[int]
     ) -> None:
         ngram_id -= 1
         # assert(ngram_id < ngram_indexes_.size());
@@ -191,8 +190,8 @@ class TfIdfVectorizer(OpRun):
         # assert(static_cast<size_t>(output_idx) < frequencies.size());
         frequencies[output_idx] += 1
 
-    def output_result(self, B: int, frequencies: List[int]) -> np.ndarray:
-        l_output_dims: List[int] = []
+    def output_result(self, B: int, frequencies: list[int]) -> np.ndarray:
+        l_output_dims: list[int] = []
         if B == 0:
             l_output_dims.append(self.output_size_)
             B = 1
@@ -243,16 +242,16 @@ class TfIdfVectorizer(OpRun):
         X: np.ndarray,
         row_num: int,
         row_size: int,
-        frequencies: List[int],
+        frequencies: list[int],
         max_gram_length=None,
         max_skip_count=None,
         min_gram_length=None,
-        mode=None,
-        ngram_counts=None,
-        ngram_indexes=None,
-        pool_int64s=None,
-        pool_strings=None,
-        weights=None,
+        mode=None,  # noqa: ARG002
+        ngram_counts=None,  # noqa: ARG002
+        ngram_indexes=None,  # noqa: ARG002
+        pool_int64s=None,  # noqa: ARG002
+        pool_strings=None,  # noqa: ARG002
+        weights=None,  # noqa: ARG002
     ) -> None:
         if len(X.shape) > 1:
             X_flat = X[row_num]

onnx/reference/ops/op_thresholded_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_tile.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_topk.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -102,7 +102,12 @@ class TopK_10(_CommonTopK):
 
 class TopK_11(_CommonTopK):
     def _run(
-        self, data, ink, axis=None, largest=None, sorted=None  # noqa: A002
+        self,
+        data,
+        ink,
+        axis=None,
+        largest=None,
+        sorted=None,  # noqa: A002
     ):  # type: ignore
         """Runtime for operator *TopK*.
 

onnx/reference/ops/op_transpose.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_trilu.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_unique.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -42,6 +42,8 @@ class Unique(OpRun):
         indices, inverse_indices, counts = _specify_int64(
             indices, inverse_indices, counts
         )
+        # numpy 2.0 has a different behavior than numpy 1.x.
+        inverse_indices = inverse_indices.squeeze()
         if len(self.onnx_node.output) == 2:
             return (y, indices)
         if len(self.onnx_node.output) == 3:

onnx/reference/ops/op_unsqueeze.py

@@ -1,7 +1,7 @@
 # Copyright (c) ONNX Project Contributors
 
 # SPDX-License-Identifier: Apache-2.0
-
+from __future__ import annotations
 
 import numpy as np
 
@@ -21,7 +21,7 @@ class Unsqueeze_1(OpRun):
             for a in axes:
                 sq = np.expand_dims(sq, axis=a)
         else:
-            raise RuntimeError(
+            raise RuntimeError(  # noqa: TRY004
                 "axes cannot be None for operator Unsqueeze (Unsqueeze_1)."
             )
         return (sq,)

onnx/reference/ops/op_upsample.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_where.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_xor.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_optimized/__init__.py

@@ -3,6 +3,7 @@
 # Copyright (c) ONNX Project Contributors
 #
 # SPDX-License-Identifier: Apache-2.0
+from __future__ import annotations
 
 from onnx.reference.ops_optimized.op_conv_optimized import Conv
 

onnx/reference/ops_optimized/op_conv_optimized.py

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