o
    …wÖin  ã                   @   sL   d dl Z d dlZd dlmZ d dlZd dlmZ dd„ Zdd„ Z	dd	„ Z
dS )
é    N)ÚPath)Ú
AutoResumec                  C   sB   t ƒ } |  tdƒ¡tdƒksJ ‚|  t d¡¡t d¡ksJ ‚d S )Nztest/checkpointsztest/checkpoints/weightsz"msc://default/tmp/test/checkpointsz*msc://default/tmp/test/checkpoints/weights)r   Úget_weights_pathr   Úmsc)Úauto_resume© r   úX/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/tests/lightning/test_resume.pyÚ!test_auto_resume_get_weights_path   s
   ÿr	   c               	   C   s^  t ƒ } d| _|  ¡ d u sJ ‚d| _|  ¡ d u sJ ‚t ¡ 9}t tj |ddd¡¡ t tj |ddd¡¡ tj |d¡| _	t
|  ¡ ƒtj |ddd¡ksOJ ‚W d   ƒ n1 sYw   Y  t ¡ C}t tj |ddd¡¡ t tj |ddd¡¡ d|› d| _	|  ¡ }t|tjƒsŽJ ‚t
|ƒtj |ddd¡ksJ ‚W d   ƒ d S 1 s¨w   Y  d S )	NFTÚcheckpointsústep=10-epoch=0-lastÚweightsÚcontextúmsc://defaultz/checkpoints)r   Úresume_if_existsÚget_context_pathÚtempfileÚTemporaryDirectoryÚosÚmakedirsÚpathÚjoinÚresume_from_directoryÚstrÚ
isinstancer   r   ©r   Útmpdirr   r   r   r   Ú!test_auto_resume_get_context_path    s*   
ÿü
	 "úr   c               	   C   s`  t ƒ } d| _|  ¡ d u sJ ‚d| _|  ¡ d u sJ ‚t ¡ :}t tj |ddd¡¡ t tj |ddd¡¡ tj |dd¡| _	t
|  ¡ ƒtj |ddd¡ksPJ ‚W d   ƒ n1 sZw   Y  t ¡ C}t tj |ddd¡¡ t tj |ddd¡¡ d|› d| _	|  ¡ }t|tjƒsJ ‚t
|ƒtj |ddd¡ksžJ ‚W d   ƒ d S 1 s©w   Y  d S )	NFTr
   r   r   r   r   z!/checkpoints/step=10-epoch=0-last)r   r   Úget_trainer_ckpt_pathr   r   r   r   r   r   Úresume_from_pathr   r   r   r   r   r   r   r   Ú&test_auto_resume_get_trainer_ckpt_path<   s*   
ÿü
	 "úr   )r   r   Úpathlibr   Úmultistorageclientr   Únemo.lightning.resumer   r	   r   r   r   r   r   r   Ú<module>   s   