a  dN @sddlZddlZddlmZmZddlmZmZddlm Z eedddZ e e de ejd de _eedd d d Zeedd d dZeedd ddZe eZe eZe eZeejedZeeddddZeeddddZdS)N)partialwraps)AnyCallable)_logger)fnreturncs"ttttdfdd }|S)zdCall a function only on rank 0 in distributed settings. Meant to be used as an decorator. argskwargsrcstjdkr|i|SdS)Nr)rank_zero_onlyrankr r rf/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchmetrics/utilities/prints.py wrapped_fns z"rank_zero_only..wrapped_fn)rr)rrrrrr sr r Z LOCAL_RANKr cOstj|i|dSN)warningswarnrrrr_warn*srcOstj|i|dSr)loginforrrr_info.srcOstj|i|dSr)rdebugrrrr_debug2sr)category)namedomainrcCs td|d|d|ddS)zJWarn user that he is importing class from location it has been deprecated. Importing `zI` from `torchmetrics` was deprecated and will be removed in 2.0. Import `` from `torchmetrics. ` instead.N_future_warningrrrrr_deprecated_root_import_class<sr%cCs td|d|d|ddS)zMWarn user that he is importing function from location it has been deprecated.rzT` from `torchmetrics.functional` was deprecated and will be removed in 2.0. Import `r r!Nr"r$rrr_deprecated_root_import_funcDsr&)osr functoolsrrtypingrrZ torchmetricsrrr getattrintenvirongetr rrrZrank_zero_debugZrank_zero_infoZrank_zero_warnr FutureWarningr#strr%r&rrrrs