Spaces:
Running
Running
File size: 304 Bytes
c61ccee |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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
|