o ni @sdZddlZddlmZmZmZddlZddlmZddl m Z ddl m Z GdddZ  dd eeeeefd eeeeefd dfd d ZdddZdS)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") N)OptionalSequenceUnion)_REQUIRE_LEGACY_SERVICE)UnsupportedError) wandb_runc@sxeZdZUdZeeed<deeeefddfddZddd Z dd d Z dd d Z dddZ dddZ dddZdS) _RequireszInternal feature class. _featuresfeaturesreturnNcCs(t|tr t|g|_dSt||_dSN) isinstancestrtupler )selfr rU/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/wandb_require.py__init__sz_Requires.__init__cCsdSr rrrrrrequire_requiresz_Requires.require_requirecCs tjt_tjt_tjjtj_dSr ) wandb _teardownteardown_attachattachrRun_detachdetachrrrr_require_service"sz_Requires._require_servicecCs |dSr )rrrrrrequire_service's z_Requires.require_servicecCstddS)NzG`wandb.require('core')` is redundant as it is now the default behavior.)rtermwarnrrrr require_core*sz_Requires.require_corecCsdtjt<dS)Ntrue)osenvironrrrrrrequire_legacy_service/sz _Requires.require_legacy_servicecCsd}|jD]1}|ddd}|ddd}d|dd}t||d }|s3d |}t|q|q|rBtd t|d S) z-Call require_* method for supported features.@r:z require_{}-_Nz#require() unsupported requirement: z8Supported requirements are: `legacy-service`, `service`.)r splitformatreplacegetattrrr r)r last_message feature_item full_featurefeaturefunc_strfuncrrrapply2s     z_Requires.applyr N)__name__ __module__ __qualname____doc__rr__annotations__rrrrrr!r%r6rrrrrs       r requirement experimentr cCs&|p|}|sdSt|d}|dS)zIndicate which experimental features are used by the script. Args: requirement: (str or list) Features to require experiment: (str or list) Features to require Raises: wandb.errors.UnsupportedError: if not supported N)r )rr6)r=r>r frrrrequireEs    r@cCs tddS)zMOn wandb import, setup anything needed based on parent process require calls.serviceN)r@rrrr_import_module_hookZs rB)NNr7)r;r#typingrrrr wandb.envr wandb.errorsr wandb.sdkrrrr@rBrrrrs$    1