o xi @sRdZddlmZddlmZddlZddlmZGdddZ  ddd dZ dS)a&Feature Flags Module. This module implements a feature flag system for the wandb library to require experimental features and notify the user when features have been deprecated. Example: import wandb wandb.require("wandb-service@beta") wandb.require("incremental-artifacts@beta") ) annotations)IterableN)UnsupportedErrorc@sLeZdZUdZded<ddd Zdd d Zdd d ZdddZdddZ dS) _RequireszInternal feature class.ztuple[str, ...] _featuresfeaturesstr | Iterable[str]returnNonecCs(t|tr t|g|_dSt||_dSN) isinstancestrtupler)selfrrK/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/wandb_require.py__init__sz_Requires.__init__cCsdSr rrrrrrequire_requiresz_Requires.require_requirecCtddS)NzH`wandb.require('service')` is a no-op as it is now the default behavior.wandbtermwarnrrrrrequire_service!sz_Requires.require_servicecCr)NzE`wandb.require('core')` is a no-op as it is now the default behavior.rrrrr require_core)sz_Requires.require_corecCs~d}|jD]1}|ddd}|ddd}d|dd}t||d }|s3d |}t|q|q|r=t|d S) z-Call require_* method for supported features.@r:z require_{}-_Nz#require() unsupported requirement: )rsplitformatreplacegetattrrrr)r last_message feature_item full_featurefeaturefunc_strfuncrrrapply0s    z_Requires.applyN)rrr r )r r ) __name__ __module__ __qualname____doc____annotations__rrrrr+rrrrrs     r requirementstr | Iterable[str] | None experimentr r cCs&|p|}|sdSt|d}|dS)aIndicate which experimental features are used by the script. This should be called before any other `wandb` functions, ideally right after importing `wandb`. Args: requirement: The name of a feature to require or an iterable of feature names. experiment: An alias for `requirement`. Raises: wandb.errors.UnsupportedError: If a feature name is unknown. N)r)rr+)r1r3rfrrrrequireBs   r5)NN)r1r2r3r2r r ) r/ __future__rcollections.abcrr wandb.errorsrrr5rrrrs  /