20 lines
452 B
Python
20 lines
452 B
Python
# Copyright (c) ONNX Project Contributors
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
from __future__ import annotations
|
|
|
|
import dataclasses
|
|
from typing import TYPE_CHECKING, Any, Callable
|
|
|
|
if TYPE_CHECKING:
|
|
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]
|