o -wi3@sUddlZddlmZddlmZmZmZmZmZededefdZ ddiZ ee e fe d <d e d e fd d Z GdddZdS)N)wraps)AnyCallableDictTypeVarcastFuncT.)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_varrr \/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/wandb_require_helpers.pywrappers  z'requires..deco..wrapper)rrrr)rrrr )rrdecos  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%r rrrrrs   r)r functoolsrtypingrrrrrrr str__annotations__rrrrrrs