1import dataclasses 2from typing import Callable, Optional 3 4from torch._guards import GuardsSet 5 6from .types import GuardFail 7 8 9@dataclasses.dataclass 10class Hooks: 11 guard_export_fn: Optional[Callable[[GuardsSet], None]] = None 12 guard_fail_fn: Optional[Callable[[GuardFail], None]] = None 13