o ni*@sdZddlZddlZddlmZmZmZddlZddlZddl m Z m Z ddl m ZddlmZdd lmZdd lmZddlm Z d%d eed eddfddZ       d&deeddeedeed eedeedeededefddZGdddejZGdddZddddddddddd deeddeedeed eedeedeed eed!eed"eefd#d$ZdS)'zZLog in to Weights & Biases. This authenticates your machine to log data to your account. N)LiteralOptionalTuple)AuthenticationError UsageError)Settings) InternalApi)Api)apikeyFhostcloudreturncCsxt}|dks |dur(|r(|jddddtjtr&|jddddSdS|r:|d}|jd|ddddSdS)zWrite the host parameter to the global settings file. This takes the parameter from wandb.login or wandb login for use by the application's APIs. zhttps://api.wandb.aiNbase_urlT)globallypersist)r/) r clear_settingospathexists OldSettings _local_pathrstrip set_setting)r r_apirS/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/wandb_login.py_handle_host_wandb_settings r anonymous)mustallowneverkeyreloginforcetimeoutverifyc Cspt|tjjr dSt||||||d}|r2ddlm}|jj } | dus(J| j j } | s2t d|r6dSdS)a^Set up W&B login credentials. By default, this will only store credentials locally without verifying them with the W&B server. To verify credentials, pass `verify=True`. Args: anonymous: (string, optional) Can be "must", "allow", or "never". If set to "must", always log a user in anonymously. If set to "allow", only create an anonymous user if the user isn't already logged in. If set to "never", never log a user anonymously. Default set to "never". key: (string, optional) The API key to use. relogin: (bool, optional) If true, will re-prompt for API key. host: (string, optional) The host to connect to. force: (bool, optional) If true, will force a relogin. timeout: (int, optional) Number of seconds to wait for user input. verify: (bool) Verify the credentials with the W&B server. Returns: bool: if key is configured Raises: AuthenticationError - if api_key fails verification with the server UsageError - if api_key cannot be configured and no tty Tr r$r%r r&r'r ) wandb_setupNz=API key verification failed. Make sure your API key is valid.F) rwandbsetup _settings_noop_loginr* _WandbSetup _instance_server_viewerr) r r$r%r r&r'r( configuredr* singletonviewerrrrlogin(s*#    r8c@seZdZdZdZdZdZdS) ApiKeyStatusr rN)__name__ __module__ __qualname__VALIDNOTTYOFFLINEDISABLEDrrrrr9es r9c@seZdZddZddddddddeeddeedeed eed eed eed dfd dZ d efddZ ddZ ddZ ded dfddZ ded dfddZd efddZd(ddZded dfdd Zejfdeed!ed dfd"d#Zd eeeeffd$d%Zd(d&d'ZdS)) _WandbLogincCs.d|_d|_d|_d|_d|_d|_d|_dSN)r-_backend_silent_entity_wl_key_reloginselfrrr__init__ms z_WandbLogin.__init__Nr)r r"r!r#r$r%r r&r'rcCsL||_|||||d}tjtjdidd|Dd|_|jj|_dS)z:Updates login-related settings on the global setup object.)r api_keyrr& login_timeoutcSsi|] \}}|dur||qSrDr).0kvrrr sz%_WandbLogin.setup..settingsNr)rJr+r,ritemsrHrVr-)rLr r$r%r r&r'login_settingsrrrr,vs  z_WandbLogin.setupcCstj|jdduS)z5Returns whether an API key is set or can be inferred.rUN)r rOr-rKrrris_apikey_configuredsz _WandbLogin.is_apikey_configuredcCs |jjduSrD)r-identity_token_filerKrrrshould_use_identity_tokens z%_WandbLogin.should_use_identity_tokencC ||_dSrD)rE)rLbackendrrr set_backend z_WandbLogin.set_backendsilentcCr\rD)rF)rLr`rrr set_silentr_z_WandbLogin.set_silententitycCr\rD)rG)rLrbrrr set_entityr_z_WandbLogin.set_entitycCs4|}|jjs |jr d}|sdS|js||S)zReturns whether the user is logged in (i.e. an API key exists). If the user is logged in, this also prints an informational message. F)rYr-r%rJrF_print_logged_in_message)rLapikey_configuredrrrr8sz_WandbLogin.logincCs|j}|r6|jp|j}d}|r)||jvr)||kr)dtj|ddd}dtj|dd|}nd}dtjd d d d }tj|d |dddS)z5Prints a message telling the user they are logged in.r0z (yellow)fg)zCurrently logged in as: zW&B API key is configuredzUse z`wandb login --relogin`T)boldz to force reloginz. F)repeatN) rH _get_usernamerG _get_entity _get_teamsclickstyler+termlog)rLusernamerb entity_strlogin_state_strlogin_info_strrrrrds   z$_WandbLogin._print_logged_in_messagecCs<|jjr |jjs tdt|j|||||_dS)z:Saves the API key and updates the the global setup object.zIf you're specifying your api key in code, ensure this code is not shared publicly. Consider setting the WANDB_API_KEY environment variable, or running `wandb login` from the command line.N) r- _notebookr`r+termwarnr write_keyupdate_sessionrI)rLr$rrrconfigure_api_keys  z_WandbLogin.configure_api_keystatuscCsjt}|tjkrtdd}n|tjkrtdd}n|r t|d}|jj||jjjs3|j dSdS)zUpdates mode and API key settings on the global setup object. If we're online, this also pulls in user settings from the server. offline)modedisabled)rON) dictr9rArBrHr-update_from_dictrV_offline_update_user_settings)rLr$rzrXrrrrxs      z_WandbLogin.update_sessionc Cst|j} ztj|j||jr|jjnd|jr|jjndd}Wn,ty:}zt|jdWYd}~qd}~wt yLt ddt j fYSw|durVdt j fS|s]dt jfS|t jfS)NT)api no_offline no_createrz"W&B disabled due to login timeout.F)r r-r prompt_api_keyr& ValueErrorr+ termerrorargs TimeoutErrorrpr9rBr@rAr?)rLrr$errr_prompt_api_keys,        z_WandbLogin._prompt_api_keycCsJ|\}}|tjkr|jjrdnd}td||j||d||_dS)z1Updates the global API key by prompting the user.zwandb login [your_api_key]zwandb.login(key=[your_api_key])z&api_key not configured (no-tty). call )rzN)rr9r@r-x_cli_only_moderrxrI)rLr$rz directiverrrrs    z_WandbLogin.prompt_api_key)rN)r<r=r>rMrrstrboolintr,rYr[r^rarcr8rdryr9r?rxrrrrrrrrClsR     rC) r r$r%r r&r'rErF_disable_warningrGrNrFrrGc Cstjdur|s tddSt} |r| ||r| || r&| | | j||||||d| jj r7dStj rHtj sHt ddS| rNdS| } |rY| || r]| S|sc| | jpgdS)Nz7Calling wandb.login() after wandb.init() has no effect.Tr)FzQTo use W&B in kaggle you must enable internet in the settings panel on the right.)r+runrvrCr^rarcr,r-rutil _is_kaggle _has_internetrr[r8ryrrI) r r$r%r r&r'rErFrrGwlogin logged_inrrrr/sH        r/)F)NNNNNNF)__doc__enumrtypingrrrrnr+ wandb.errorsrrwandb.old.settingsrrapisr internal.internal_apir libr wandb_settingsrrrrr8Enumr9rCr/rrrrs       =,