o
    mi                      @   s|   d dl Z d dlZd dlZd dlZd dlmZmZ ddlmZ	 dee
e
f fddZdeee
e
f ee
e
f f fdd	ZdS )
    N)DictTuple   )filesreturnc                  C   sD   t  } tjtjr ttjD ]}| dd\}}|| |< q| S )z2We read our api_key from secrets.env in SageMaker.=r   )	dictospathexistssm_files
SM_SECRETSopenstripsplit)env_dictlinekeyval r   c/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/integration/sagemaker/resources.pyparse_sm_secrets
   s   
r   c               	   C   sv   t  } td}|r-tdd u r-ddd tdD }d||tdt g| d	< td| d
< t }| |fS )NTRAINING_JOB_NAMEWANDB_RUN_ID c                 s   s"    | ]}t tjtj V  qd S )N)secretschoicestringascii_lowercasedigits).0_r   r   r   	<genexpr>   s    
z%parse_sm_resources.<locals>.<genexpr>   -CURRENT_HOSTrun_id	run_group)r   r	   getenvjoinrangesocketgethostnamer   )run_dictr&   suffixr   r   r   r   parse_sm_resources   s   

r/   )r	   r   r+   r   typingr   r   r   r   r   strr   r/   r   r   r   r   <module>   s    *