You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
13 lines
292 B
13 lines
292 B
5 months ago
|
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
|