JSTprove 1.2.0__py3-none-macosx_11_0_arm64.whl → 1.4.0__py3-none-macosx_11_0_arm64.whl

python/tests/onnx_quantizer_tests/test_registered_quantizers.py

@@ -49,11 +49,10 @@ def validate_quantized_node(node_result: onnx.NodeProto, op_type: str) -> None:
     assert node_result.output, f"Missing outputs for {op_type}"
 
     try:
-        # Create a minimal model with custom opset for validation
+        # Create a minimal graph with dummy IOs to satisfy ONNX requirements
         temp_graph = onnx.GraphProto()
         temp_graph.name = "temp_graph"
 
-        # Add dummy inputs/outputs to satisfy graph requirements
         for inp in node_result.input:
             if not any(vi.name == inp for vi in temp_graph.input):
                 temp_graph.input.append(
@@ -63,6 +62,7 @@ def validate_quantized_node(node_result: onnx.NodeProto, op_type: str) -> None:
                         [1],
                     ),
                 )
+
         for out in node_result.output:
             if not any(vi.name == out for vi in temp_graph.output):
                 temp_graph.output.append(
@@ -74,12 +74,16 @@ def validate_quantized_node(node_result: onnx.NodeProto, op_type: str) -> None:
                 )
 
         temp_graph.node.append(node_result)
-        temp_model = onnx.helper.make_model(temp_graph)
-        custom_domain = onnx.helper.make_operatorsetid(
-            domain="ai.onnx.contrib",
-            version=1,
+
+        # Explicit opset imports for default and contrib domains
+        temp_model = onnx.helper.make_model(
+            temp_graph,
+            opset_imports=[
+                onnx.helper.make_opsetid("", 22),
+                onnx.helper.make_opsetid("ai.onnx.contrib", 1),
+            ],
         )
-        temp_model.opset_import.append(custom_domain)
+
         onnx.checker.check_model(temp_model)
     except onnx.checker.ValidationError as e:
         pytest.fail(f"ONNX node validation failed for {op_type}: {e}")
@@ -117,5 +121,10 @@ def test_registered_quantizer_quantize(
         for node_result in result:
             validate_quantized_node(node_result, op_type)
     else:
-        assert result.input, f"Missing inputs for {op_type}"
+        if inputs:
+            # Only assert if this op actually requires inputs
+            assert (
+                result.input
+            ), f"Missing inputs for {op_type}; required_inputs={inputs}"
+
         validate_quantized_node(result, op_type)