o zi @sdZddlZddlmZddlmZmZmZddlm Z ej de dej Z dd e d dfd d Z dd eee fdeede dedee d e f ddZee_ GdddeZded efddZdS)zWarning-related utilities.N)Path)OptionalTypeUnion)LightningDeprecationWarningdefault)categorymodulereturncCstjd|tddS)auIgnore warnings of the category ``PossibleUserWarning`` from Lightning. For more granular control over which warnings to ignore, use :func:`warnings.filterwarnings` directly. Args: module: Name of the module for which the warnings should be ignored (e.g., ``'pytorch_lightning.strategies'``). Default: Disables warnings from all modules. ignore)r rN)warningsfilterwarningsPossibleUserWarning)r rW/home/ubuntu/.local/lib/python3.10/site-packages/lightning_fabric/utilities/warnings.pydisable_possible_user_warningss rmessagerfilenamelinenolinecCs2tt|r|d|d|dSt|||||S)zhCustom formatting that avoids an extra line in case warnings are emitted from the `rank_zero`-functions.:z:  )_is_path_in_lightningr_default_format_warning)rrrrrrrr_custom_format_warning(s rc@seZdZdZdS)rz'Warnings that could be false positives.N)__name__ __module__ __qualname____doc__rrrrr5srpathcCsdt|vS)zJNaive check whether the path looks like a path from the lightning package. lightning)strabsolute)r rrrr9sr)r )N)rr pathlibrtypingrrr$lightning_fabric.utilities.rank_zeror simplefilter formatwarningrr"rWarningintr UserWarningrboolrrrrrs2