o -wi0,@sdZddlZddlZddlmZmZmZddlZddlm Z ddl Z ddl m Z m Z ddlmZddlmZdd lmZd d lmZd d lmZd(deededdfddZ       d)deeddeedeedeedeedeededeedefddZGdddejZGdd d Z ddddddd d!ddd" deed#deedeedeedeedeededed$eed%eedefd&d'Z!dS)*zZLog in to Weights & Biases. This authenticates your machine to log data to your account. N)LiteralOptionalTuple)ConnectionError)AuthenticationError UsageError)Settings) wandb_setup) 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_urlTgloballypersist)r/) r clear_settingospathexists OldSettings _local_pathrstrip set_setting)rr_apir R/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/wandb_login.py_handle_host_wandb_settings r" anonymousmustallowneverkeyreloginforcetimeoutverifyreferrerc Cs t|t||||||||dS)aSet 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. referrer: (string, optional) The referrer to use in the URL login request. 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 r#r(r)rr*r+r,r-)r"_loginr.r r r!login)s%r0c@seZdZdZdZdZdZdS) ApiKeyStatusr r N)__name__ __module__ __qualname__VALIDNOTTYOFFLINEDISABLEDr r r r!r1[s r1c@s eZdZ      ddeeddeedeedeedeedeef d d Zdd d Z d efddZ dddZ ded dfddZ e jfdeede d dfddZ d deed eeee ffddZ d deed eeee ffddZded dfddZdS)! _WandbLoginNr#r$r*rr(r)r+cCsF||_|||||d}|dk|_t|_|jj||jj|_dS)N)r#api_keyrr* login_timeoutr%)_relogin is_anonymousr singleton _wandb_setupsettingsupdate_from_dict _settings)selfr#r*rr(r)r+login_settingsr r r!__init__cs   z_WandbLogin.__init__rcCs6t}|jr|jddddddS|jdddddS)Nr#r%Tr)r r?rr)rEapir r r! _update_global_anonymous_setting{sz,_WandbLogin._update_global_anonymous_settingcCstj|jdduS)z5Returns whether an API key is set or can be inferred.rBN)rr<rD)rEr r r!is_apikey_configuredsz _WandbLogin.is_apikey_configuredcCs|j}|rK|jjrdtj|jjddnd}|jjp!|j}d}|r<||jvr<||krtj s>t ddS| jj rDdSd} d} |durbt j| jd}|rZ|sZd} n| j|d\}} |ri| || s{| || j|| d | |r|s| |duS) Nz7Calling wandb.login() after wandb.init() has no effect.T)r#r*rr(r)r+z'Unable to verify login in offline mode.FzQTo use W&B in kaggle you must enable internet in the settings panel on the right.rJ)r-)ri)rYrunrcr;rD_nooprorzutil _is_kaggle _has_internetrvidentity_token_filerr<rtrrhrqrIr`) r#r(r)rr*r+r,r-rrwloginkey_is_pre_configured key_statusr r r!r/ sP      r/)F)NNNNNNFN)"__doc__enumrtypingrrrrTrequests.exceptionsrrY wandb.errorsrrwandb.old.settingsrr wandb.sdkr apisr internal.internal_apir librrfrr"rr0Enumr1r;r/r r r r!s        2*