Kano001's picture
Upload 5252 files
c61ccee verified
raw
history blame
304 Bytes
import dataclasses
from typing import Callable, Optional
from torch._guards import GuardsSet
from .types import GuardFail
@dataclasses.dataclass
class Hooks:
guard_export_fn: Optional[Callable[[GuardsSet], None]] = None
guard_fail_fn: Optional[Callable[[GuardFail], None]] = None