o ic@sPdZddlZddlmZmZddlmZddlmZm Z ej GdddZ dS) aHook system for Dynamo's guard functionality. This module provides a way to register callback functions that are triggered during guard-related operations. The Hooks class manages two types of hook functions: - guard_export_fn: Called when guards need to be exported, taking a GuardsSet as input - guard_fail_fn: Called when a guard check fails, taking a GuardFail object as input These hooks enable customization of guard export and failure handling behaviors. N)CallableOptional) GuardsSet) GuardFailGuardFilterEntryc@sdeZdZUdZeeegdfed<dZeee gdfed<dZ eee e ge e fed<dS)HooksNguard_export_fn guard_fail_fnguard_filter_fn)__name__ __module__ __qualname__r rrr__annotations__r rr listrboolrrP/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/torch/_dynamo/hooks.pyrs &r) __doc__ dataclassestypingrr torch._guardsrtypesrr dataclassrrrrrs