mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
7abd035b2f
I'm not sure how lintrunner worked without this lol. Signed-off-by: Edward Z. Yang <ezyang@fb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/90179 Approved by: https://github.com/albanD, https://github.com/voznesenskym
10 lines
250 B
Python
10 lines
250 B
Python
import dataclasses
|
|
|
|
from typing import Callable, Optional, Set, Tuple
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class Hooks:
|
|
guard_export_fn: Optional[Callable[[Set["Guard"]], None]] = None
|
|
guard_fail_fn: Optional[Callable[[Tuple["GuardFail"]], None]] = None
|