19 lines
415 B
Python
19 lines
415 B
Python
# Copyright (c) ONNX Project Contributors
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
from __future__ import annotations
|
|
|
|
import dataclasses
|
|
from typing import Any, Callable
|
|
|
|
from onnx import ModelProto, NodeProto
|
|
|
|
# A container that hosts the test function and the associated
|
|
# test item (ModelProto)
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class TestItem:
|
|
func: Callable[..., Any]
|
|
proto: list[ModelProto | NodeProto | None]
|