onnx 1.17.0__cp39-cp39-macosx_12_0_universal2.whl → 1.18.0__cp39-cp39-macosx_12_0_universal2.whl

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_expanded/test_data_set_0/input_1.pb

@@ -0,0 +1 @@
+
B	cos_cacheJ`J٣>aU�>�?��T?�!?	f_?�

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_expanded/test_data_set_0/input_2.pb

@@ -0,0 +1 @@
+
B	sin_cacheJ`a�>>d�q?3S=?k�>a�h>;�>��m=�k�>���>�C2?�h�>�7>+�<-��=��-?�J�>A]	?@�e?چ}?>^>~�)?1҆>I,�<%B?

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_interleaved/test_data_set_0/input_1.pb

@@ -0,0 +1 @@
+
B	cos_cacheJ`J٣>aU�>�?��T?�!?	f_?�

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_interleaved/test_data_set_0/input_2.pb

@@ -0,0 +1 @@
+
B	sin_cacheJ`a�>>d�q?3S=?k�>a�h>;�>��m=�k�>���>�C2?�h�>�7>+�<-��=��-?�J�>A]	?@�e?چ}?>^>~�)?1҆>I,�<%B?

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_interleaved_expanded/test_data_set_0/input_1.pb

@@ -0,0 +1 @@
+
B	cos_cacheJ`J٣>aU�>�?��T?�!?	f_?�

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_interleaved_expanded/test_data_set_0/input_2.pb

@@ -0,0 +1 @@
+
B	sin_cacheJ`a�>>d�q?3S=?k�>a�h>;�>��m=�k�>���>�C2?�h�>�7>+�<-��=��-?�J�>A]	?@�e?چ}?>^>~�)?1҆>I,�<%B?

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_rotary_dim/test_data_set_0/input_1.pb

@@ -0,0 +1 @@
+
B	cos_cacheJ`J٣>aU�>�?��T?�!?	f_?�

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_rotary_dim/test_data_set_0/input_2.pb

@@ -0,0 +1 @@
+
B	sin_cacheJ`a�>>d�q?3S=?k�>a�h>;�>��m=�k�>���>�C2?�h�>�7>+�<-��=��-?�J�>A]	?@�e?چ}?>^>~�)?1҆>I,�<%B?

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_rotary_dim_expanded/test_data_set_0/input_1.pb

@@ -0,0 +1 @@
+
B	cos_cacheJ`J٣>aU�>�?��T?�!?	f_?�

onnx/backend/test/data/node/test_rotary_embedding_no_position_ids_rotary_dim_expanded/test_data_set_0/input_2.pb

@@ -0,0 +1 @@
+
B	sin_cacheJ`a�>>d�q?3S=?k�>a�h>;�>��m=�k�>���>�C2?�h�>�7>+�<-��=��-?�J�>A]	?@�e?چ}?>^>~�)?1҆>I,�<%B?

onnx/backend/test/data/node/test_sinh/test_data_set_0/output_0.pb

@@ -1 +1 @@
-
B
yJ�
%E5@hd�>�B�?͹�@�,J@����6��?
+
B
yJ�
$E5@gd�>�B�?̹�@�,J@����6��?

onnx/backend/test/data/node/test_tan/test_data_set_0/output_0.pb

@@ -1 +1 @@
-
B
yJ�
R������>�R�?=����JQ�f�����?-/�k%ԽF��>��>��A[�s?pm�=�u�>5z�>&PAzU����>s꒿��*?�D?!�?��j��I�����̎;=]�A��T�A�0A��>�h�>�P��P@@!���6k!>��4@��%@��оD����e޿
+
B
yJ�
R������>�R�?=����JQ�f�����?-/�j%ԽF��>�>��A[�s?pm�=�u�>5z�>&PAzU����>s꒿��*?�D? �?��j��I�����̎;=]�A��T�A�0A��>�h�>�P��Q@@!���6k!>��4@��%@��оD����e޿

onnx/backend/test/report/__init__.py

@@ -3,13 +3,17 @@
 # SPDX-License-Identifier: Apache-2.0
 from __future__ import annotations
 
-from typing import Any, Sequence
+from typing import TYPE_CHECKING, Any
 
-import _pytest
 import pytest
 
 from onnx.backend.test.report.coverage import Coverage
 
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
+    import _pytest
+
 _coverage = Coverage()
 _marks: dict[str, Sequence[Any]] = {}
 
@@ -36,9 +40,10 @@ def pytest_runtest_logreport(report: Any) -> None:
         _add_mark(mark, "passed")
 
 
-@pytest.hookimpl(trylast=True)  # type: ignore
+@pytest.hookimpl(trylast=True)
 def pytest_terminal_summary(
-    terminalreporter: _pytest.terminal.TerminalReporter, exitstatus: int  # noqa: ARG001
+    terminalreporter: _pytest.terminal.TerminalReporter,
+    exitstatus: int,  # noqa: ARG001
 ) -> None:
     for mark in _marks.values():
         _add_mark(mark, "loaded")

onnx/backend/test/report/coverage.py

@@ -153,12 +153,8 @@ class Coverage:
                 if schema.support_level == defs.OpSchema.SupportType.EXPERIMENTAL:
                     experimental.append(schema.name)
         all_ops.sort()
-        nodes_path = os.path.join(
-            str(os.environ.get("CSVDIR")), "nodes.csv"  # type: ignore
-        )  # type: ignore
-        models_path = os.path.join(
-            str(os.environ.get("CSVDIR")), "models.csv"  # type: ignore
-        )  # type: ignore
+        nodes_path = os.path.join(str(os.environ.get("CSVDIR")), "nodes.csv")
+        models_path = os.path.join(str(os.environ.get("CSVDIR")), "models.csv")
         existing_nodes: OrderedDict[str, dict[str, str]] = OrderedDict()
         existing_models: OrderedDict[str, dict[str, str]] = OrderedDict()
         frameworks: list[str] = []
@@ -219,7 +215,7 @@ class Coverage:
             # Consider both buckets
             num_models = 0
             for bucket in self.models:
-                for model in self.models[bucket]:  # type: ignore
+                for model in self.models[bucket]:
                     # Both analyze and run the model on the backend
                     num_covered = 0
                     for node in self.models[bucket][model].node_coverages:
@@ -251,14 +247,14 @@ class Coverage:
                 f"{len(self.models['passed'])}/{num_models} model tests passed"
             )
             summaries["Model"] = "Summary"
-            for model in existing_models:  # type: ignore
+            for model in existing_models:
                 existing_models[model]["Model"] = model
                 model_writer.writerow(existing_models[model])
             model_writer.writerow(summaries)
         with open(
-            os.path.join(str(os.environ.get("CSVDIR")), "metadata.csv"),  # type: ignore
+            os.path.join(str(os.environ.get("CSVDIR")), "metadata.csv"),
             "w",
-        ) as metadata_file:  # type: ignore
+        ) as metadata_file:
             metadata_writer = csv.writer(metadata_file)
             metadata_writer.writerow(
                 ["Latest Update", datetime.datetime.now().isoformat().replace("T", " ")]

onnx/backend/test/runner/__init__.py

@@ -14,7 +14,8 @@ import tempfile
 import time
 import unittest
 from collections import defaultdict
-from typing import Any, Callable, Iterable, Pattern, Sequence
+from re import Pattern
+from typing import TYPE_CHECKING, Any, Callable
 from urllib.request import urlretrieve
 
 import numpy as np
@@ -22,11 +23,15 @@ import numpy as np
 import onnx
 import onnx.reference
 from onnx import ONNX_ML, ModelProto, NodeProto, TypeProto, ValueInfoProto, numpy_helper
-from onnx.backend.base import Backend
-from onnx.backend.test.case.test_case import TestCase
 from onnx.backend.test.loader import load_model_tests
 from onnx.backend.test.runner.item import TestItem
 
+if TYPE_CHECKING:
+    from collections.abc import Iterable, Sequence
+
+    from onnx.backend.base import Backend
+    from onnx.backend.test.case.test_case import TestCase
+
 
 class BackendIsNotSupposedToImplementIt(unittest.SkipTest):
     pass
@@ -214,7 +219,12 @@ class Runner:
                         raise AssertionError(f"{ref_outputs[i]} != {outputs[i]}")
                     continue
                 np.testing.assert_equal(outputs[i].dtype, ref_outputs[i].dtype)
-                if ref_outputs[i].dtype == object:  # type: ignore[attr-defined]
+                np.testing.assert_array_equal(
+                    outputs[i].shape,
+                    ref_outputs[i].shape,
+                    err_msg=f"Output {i} has incorrect shape",
+                )
+                if ref_outputs[i].dtype == object:
                     np.testing.assert_array_equal(outputs[i], ref_outputs[i])
                 else:
                     np.testing.assert_allclose(
@@ -226,27 +236,21 @@ class Runner:
     def download_model(
         cls,
         model_test: TestCase,
-        model_dir: str,
         models_dir: str,
     ) -> None:
-        # On Windows, NamedTemporaryFile can not be opened for a
-        # second time
-        del model_dir
-        download_file = tempfile.NamedTemporaryFile(delete=False)
-        try:
-            download_file.close()
-            assert model_test.url
-            print(
-                f"Start downloading model {model_test.model_name} from {model_test.url}"
-            )
-            urlretrieve(model_test.url, download_file.name)
-            print("Done")
-            onnx.utils._extract_model_safe(download_file.name, models_dir)
-        except Exception as e:
-            print(f"Failed to prepare data for model {model_test.model_name}: {e}")
-            raise
-        finally:
-            os.remove(download_file.name)
+        with tempfile.TemporaryDirectory() as tmpdir:
+            try:
+                assert model_test.url
+                print(
+                    f"Start downloading model {model_test.model_name} from {model_test.url}"
+                )
+                filename = os.path.join(tmpdir, "file")
+                urlretrieve(model_test.url, filename)
+                print("Done")
+                onnx.utils._extract_model_safe(filename, models_dir)
+            except Exception as e:
+                print(f"Failed to prepare data for model {model_test.model_name}: {e}")
+                raise
 
     @classmethod
     def prepare_model_data(cls, model_test: TestCase) -> str:
@@ -267,9 +271,7 @@ class Runner:
                     break
             os.makedirs(model_dir)
 
-            cls.download_model(
-                model_test=model_test, model_dir=model_dir, models_dir=models_dir
-            )
+            cls.download_model(model_test=model_test, models_dir=models_dir)
         return model_dir
 
     def _add_test(
@@ -292,7 +294,7 @@ class Runner:
                     f'Duplicated test name "{device_test_name}" in category "{category}"'
                 )
 
-            @unittest.skipIf(  # type: ignore
+            @unittest.skipIf(
                 not self.backend.supports_device(device),
                 f"Backend doesn't support device {device}",
             )

onnx/backend/test/runner/item.py

@@ -4,9 +4,10 @@
 from __future__ import annotations
 
 import dataclasses
-from typing import Any, Callable
+from typing import TYPE_CHECKING, Any, Callable
 
-from onnx import ModelProto, NodeProto
+if TYPE_CHECKING:
+    from onnx import ModelProto, NodeProto
 
 # A container that hosts the test function and the associated
 # test item (ModelProto)

onnx/backend/test/stat_coverage.py

@@ -6,13 +6,16 @@
 from __future__ import annotations
 
 import os
-from typing import IO, Any, Sequence
+from typing import IO, TYPE_CHECKING, Any
 
 from onnx import AttributeProto, defs, load
 from onnx.backend.test.case import collect_snippets
 from onnx.backend.test.loader import load_model_tests
 from onnx.backend.test.runner import Runner
 
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
 
 def is_ml(schemas: Sequence[defs.OpSchema]) -> bool:
     return any(s.domain == "ai.onnx.ml" for s in schemas)
@@ -252,7 +255,7 @@ def gen_model_test_coverage(
 
 
 def gen_overall_test_coverage(
-    schemas: Sequence[defs.OpSchema], f: IO[Any], ml: bool  # noqa: ARG001
+    f: IO[Any],
 ) -> None:
     f.write("# Overall Test Coverage\n")
     f.write("## To be filled.\n")
@@ -276,7 +279,7 @@ def main() -> None:
         gen_outlines(f, False)
         gen_node_test_coverage(schemas, f, False)
         gen_model_test_coverage(schemas, f, False)
-        gen_overall_test_coverage(schemas, f, False)
+        gen_overall_test_coverage(f)
 
     if has_ml:
         fname = os.path.join(docs_dir, "TestCoverage-ml.md")
@@ -285,7 +288,7 @@ def main() -> None:
             gen_outlines(f, True)
             gen_node_test_coverage(schemas, f, True)
             gen_model_test_coverage(schemas, f, True)
-            gen_overall_test_coverage(schemas, f, True)
+            gen_overall_test_coverage(f)
 
 
 if __name__ == "__main__":