o xi-@s~UddlZddlmZddlmZmZmZmZededefdZddiZ e e e fe d <d e d efd d Z GdddZdS)N)wraps)AnyCallableTypeVarcastFuncT.)boundzreport-editing:v0WANDB_REQUIRE_REPORT_EDITING_V0requirement_env_var_mapping requirementreturncs*tdtdtffdd }tt|S)z7Decorate functions to gate features with wandb.require.funcr cs0tdtdtdtffdd }tt|S)Nargskwargsr cs(ts tdd|i|S)Nz4You need to enable this feature with `wandb.require(z)`)osgetenv Exception)rr)env_varr r S/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/wandb_require_helpers.pywrappers  z'requires..deco..wrapper)rrrr)r rrr )r rdecos  zrequires..deco)r rr)r rrrrrequires s rc@s.eZdZdZd ddZd ddZd dd ZdS) RequiresMixinr NcC |dSN_check_if_requirements_metselfrrr__init__! zRequiresMixin.__init__cCrrrr rrr __post_init__$r#zRequiresMixin.__post_init__cCs*t|j}t|std|jddS)Nz=You must explicitly enable this feature with `wandb.require("z)")r r rrr)r!rrrrr's   z(RequiresMixin._check_if_requirements_met)r N)__name__ __module__ __qualname__r r"r$rrrrrrs   r)r functoolsrtypingrrrrrr dictstr__annotations__rrrrrrs