Jan 8, 2024 · Jan 5, 2024 · Jan 5, 2024 · Jan 8, 2024 · Jan 8, 2024 · Jan 8, 2024
diff --git a/_unittests/ut_light_api/test_backend_export.py b/_unittests/ut_light_api/test_backend_export.py
 import sys
 import unittest
 from typing import Any, Dict, List, Optional
 from difflib import unified_diff
 from onnx_array_api.light_api import translate
 from onnx_array_api.plotting.text_plot import onnx_simple_text_plot

 verbosity = 10 if "-v" in sys.argv or "--verbose" in sys.argv else 0


 class ReferenceImplementationError(RuntimeError):
    "Fails, export cannot be compared."

    def __init__(self, model):
        self.model = model
        self.expected_sess = ExtendedReferenceEvaluator(self.model)
        self.expected_sess = ExtendedReferenceEvaluator(self.model, verbose=verbosity)

    @property
    def input_names(self):
                    f"Unable to executed code for api {api!r}\n{new_code}"
                ) from e
            export_model = locs["model"]
            ref = ExtendedReferenceEvaluator(export_model)
            ref = ExtendedReferenceEvaluator(export_model, verbose=verbosity)
            try:
                got = ref.run(names, feeds)
            except (TypeError, AttributeError) as e:
diff --git a/onnx_array_api/light_api/make_helper.py b/onnx_array_api/light_api/make_helper.py
    """
    Constructs a NodeProto.

    Args:
        op_type: The name of the operator to construct
        inputs: list of input names
        outputs: list of output names
        name: optional unique identifier for NodeProto
        doc_string: optional documentation string for NodeProto
        domain: optional domain for NodeProto.
            If it's None, we will just use default domain (which is empty)
        **kwargs (dict): the attributes of the node.  The acceptable values
            are documented in :func:`make_attribute`.

    Returns:
        NodeProto
    :param op_type: The name of the operator to construct
    :param inputs: list of input names
    :param outputs: list of output names
    :param name: optional unique identifier for NodeProto
    :param doc_string: optional documentation string for NodeProto
    :param domain: optional domain for NodeProto.
        If it's None, we will just use default domain (which is empty)
    :param kwargs: the attributes of the node.
    :return: node proto
    """
    node = NodeProto()
    node.op_type = op_type
Original file line number	Diff line number	Diff line change
		@@ -1,3 +1,4 @@
		import sys
		import unittest
		from typing import Any, Dict, List, Optional
		from difflib import unified_diff
Expand DownExpand Up		@@ -25,6 +26,8 @@
		from onnx_array_api.light_api import translate
		from onnx_array_api.plotting.text_plot import onnx_simple_text_plot

		verbosity = 10 if "-v" in sys.argv or "--verbose" in sys.argv else 0


		class ReferenceImplementationError(RuntimeError):
		"Fails, export cannot be compared."
Expand All		@@ -36,7 +39,7 @@ class ExportWrapper:

		def __init__(self, model):
		self.model = model
		self.expected_sess = ExtendedReferenceEvaluator(self.model)
		self.expected_sess = ExtendedReferenceEvaluator(self.model, verbose=verbosity)

		@property
		def input_names(self):
Expand DownExpand Up		@@ -109,7 +112,7 @@ def run(
		f"Unable to executed code for api {api!r}\n{new_code}"
		) from e
		export_model = locs["model"]
		ref = ExtendedReferenceEvaluator(export_model)
		ref = ExtendedReferenceEvaluator(export_model, verbose=verbosity)
		try:
		got = ref.run(names, feeds)
		except (TypeError, AttributeError) as e:
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -34,19 +34,15 @@ def make_node_extended(
		"""
		Constructs a NodeProto.

		Args:
		op_type: The name of the operator to construct
		inputs: list of input names
		outputs: list of output names
		name: optional unique identifier for NodeProto
		doc_string: optional documentation string for NodeProto
		domain: optional domain for NodeProto.
		If it's None, we will just use default domain (which is empty)
		**kwargs (dict): the attributes of the node. The acceptable values
		are documented in :func:`make_attribute`.

		Returns:
		NodeProto
		:param op_type: The name of the operator to construct
		:param inputs: list of input names
		:param outputs: list of output names
		:param name: optional unique identifier for NodeProto
		:param doc_string: optional documentation string for NodeProto
		:param domain: optional domain for NodeProto.
		If it's None, we will just use default domain (which is empty)
		:param kwargs: the attributes of the node.
		:return: node proto
		"""
		node = NodeProto()
		node.op_type = op_type
Expand Down