o
    -wi                     @   s6   d dl Z d dlZd dlmZ d dlmZ dddZdS )    N)env)wandb_setup.c                 C   s   t  j}tjjj|d}| pt } | t	j
|p|}|du r#td|| t	j
< ttj|dd}|  D ]\}}|| d| d q7W d   dS 1 sRw   Y  dS )a"  Write a secrets.env file with the W&B ApiKey and any additional secrets passed.

    Args:
        overrides (dict, optional): Additional environment variables to write
                                    to secrets.env
        path (str, optional): The path to write the secrets file.
    )settingsNzNCan't find W&B ApiKey, set the WANDB_API_KEY env variable or run `wandb login`zsecrets.envw=
)r   	singletonr   wandb	wandb_libapikeyapi_keydictgetr   API_KEY
ValueErroropenospathjoinitemswrite)	overridesr   r   r   current_api_keyfilekv r   ]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/integration/sagemaker/auth.pysagemaker_auth   s   


"r   )Nr   N)r   r
   r   	wandb.sdkr   r   r   r   r   r   <module>   s
    